diff --git a/PyTorch/Translation/Transformer/train.py b/PyTorch/Translation/Transformer/train.py index 98147d8a..a749e535 100644 --- a/PyTorch/Translation/Transformer/train.py +++ b/PyTorch/Translation/Transformer/train.py @@ -364,7 +364,7 @@ def _all_gather_predictions(predictions): def save_checkpoint(args, trainer, epoch_itr, val_loss): - if epoch_itr.epoch % args.save_interval == 0: + if epoch_itr.epoch % args.save_interval != 0: return if args.no_save or not distributed_utils.is_master(args): return