trigger ci

This commit is contained in:
Adrian Wälchli 2020-11-05 01:45:02 +01:00
parent 60d1fd3d50
commit 97bf461cf9
No known key found for this signature in database
GPG Key ID: 51430D866CDDAD9C
1 changed files with 1 additions and 0 deletions

View File

@ -200,6 +200,7 @@ class ModelCheckpoint(Callback):
This method runs on all ranks, it is the responsibility of `self.save_function`
to handle correct behaviour in distributed training, i.e., saving only on rank 0.
"""
epoch = trainer.current_epoch
global_step = trainer.global_step