Merge pull request #1018 from Peter-1213/master
[Transformer/PyT] Fix a logic flaw that causes checkpoints to be saved at wrong epochs in the transformer example.
This commit is contained in:
commit
8ed4627567
|
@ -364,7 +364,7 @@ def _all_gather_predictions(predictions):
|
||||||
|
|
||||||
|
|
||||||
def save_checkpoint(args, trainer, epoch_itr, val_loss):
|
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
|
return
|
||||||
if args.no_save or not distributed_utils.is_master(args):
|
if args.no_save or not distributed_utils.is_master(args):
|
||||||
return
|
return
|
||||||
|
|
Loading…
Reference in a new issue