We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1c608ae commit c99ae81Copy full SHA for c99ae81
examples/mlperf/model_train.py
@@ -79,7 +79,7 @@ def train_resnet():
79
lr_warmup_epochs = config["lr_warmup_epochs"] = getenv("WARMUP_EPOCHS", 2)
80
decay = config["decay"] = getenv("DECAY", 2e-4)
81
82
- loss_scaler = config["LOSS_SCALER"] = getenv("LOSS_SCALER", 128.0 if dtypes.default_float == dtypes.float16 else 1.0)
+ loss_scaler = config["LOSS_SCALER"] = getenv("LOSS_SCALER", 256.0 if dtypes.default_float == dtypes.float16 else 1.0)
83
84
target, achieved = getenv("TARGET", 0.759), False
85
eval_start_epoch = getenv("EVAL_START_EPOCH", 0)
0 commit comments