maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Merge pull request #1018 from Peter-1213/master

Browse source 
[Transformer/PyT] Fix a logic flaw that causes checkpoints to be saved at wrong epochs in the transformer example.
This commit is contained in:
nv-kkudrynski 2021-11-08 12:27:49 +01:00 committed by GitHub
parent 2592d5a02c 9cf6c3824b
commit 8ed4627567
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
PyTorch/Translation/Transformer/train.py
View file

@ -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