|
@ -279,7 +279,7 @@ if __name__ == '__main__': |
|
|
torch.backends.cudnn.benchmark = hparams.cudnn_benchmark |
|
|
torch.backends.cudnn.benchmark = hparams.cudnn_benchmark |
|
|
|
|
|
|
|
|
print("FP16 Run:", hparams.fp16_run) |
|
|
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("Distributed Run:", hparams.distributed_run) |
|
|
print("cuDNN Enabled:", hparams.cudnn_enabled) |
|
|
print("cuDNN Enabled:", hparams.cudnn_enabled) |
|
|
print("cuDNN Benchmark:", hparams.cudnn_benchmark) |
|
|
print("cuDNN Benchmark:", hparams.cudnn_benchmark) |
|
|