cleaned up some if statements

Lightning-AI · Jul 25, 2019 · 7e52f6e · 7e52f6e
1 parent 7e728d9
commit 7e52f6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
@@ -885,6 +885,6 @@ def __run_validation(self):
             self.prog_bar.set_postfix(**tqdm_metrics)
 
         # model checkpointing
-        if self.proc_rank == 0 and self.checkpoint_callback:
+        if self.proc_rank == 0 and self.checkpoint_callback is not None:
             print('save callback...')
             self.checkpoint_callback.on_epoch_end(epoch=self.current_epoch, logs=self.__tng_tqdm_dic)