Skip to content

Commit

Permalink
add asserts to make sure log zero only saves checkpoints
Browse files Browse the repository at this point in the history
  • Loading branch information
williamFalcon committed Jun 30, 2020
1 parent f2c7eb0 commit 7d7088b
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions pytorch_lightning/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,11 @@
from pytorch_lightning.trainer.lr_finder import TrainerLRFinderMixin
from pytorch_lightning.utilities.exceptions import MisconfigurationException
from pytorch_lightning.utilities import rank_zero_warn, parsing, rank_zero_info, rank_zero_only
import warnings

# warnings to ignore
warnings.filterwarnings('ignore', message='torch.distributed.reduce_op is deprecated, '
'please use torch.distributed.ReduceOp instead')

try:
from apex import amp
Expand Down

0 comments on commit 7d7088b

Please sign in to comment.