diff --git a/train.py b/train.py index b13f01f..786d1dd 100644 --- a/train.py +++ b/train.py @@ -279,7 +279,7 @@ if __name__ == '__main__': torch.backends.cudnn.benchmark = hparams.cudnn_benchmark print("FP16 Run:", hparams.fp16_run) - print("Dynamic Loss Scaling", hparams.dynamic_loss_scaling) + print("Dynamic Loss Scaling:", hparams.dynamic_loss_scaling) print("Distributed Run:", hparams.distributed_run) print("cuDNN Enabled:", hparams.cudnn_enabled) print("cuDNN Benchmark:", hparams.cudnn_benchmark)