Skip to content

Commit 10e92e9

Browse files
committed
bugfix: use correct hparam/score value for invalids
Signed-off-by: Jack Luar <jluar@precisioninno.com>
1 parent b7c958c commit 10e92e9

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tools/AutoTuner/src/autotuner/tensorboard_logger.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
import ray
66
from tensorboardX import SummaryWriter
77

8+
from autotuner.utils import ERROR_METRIC
9+
810
logger = logging.getLogger(__name__)
911

1012

@@ -51,7 +53,7 @@ def log_sweep_metrics(
5153
k: v if isinstance(v, (int, float, str, bool)) else str(v)
5254
for k, v in params.items()
5355
},
54-
{"hparam/score": score if score != 999999.0 else 0.0},
56+
{"hparam/score": score if score < ERROR_METRIC else 0.0},
5557
)
5658

5759
self.step += 1

0 commit comments

Comments
 (0)