From 3aa9cfc18ea88f4ef8666c17f90325f259ade719 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Mon, 15 Jul 2019 13:18:56 -0400
Subject: [PATCH] made checkpoint callback optional

---
 pytorch_lightning/models/trainer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 052a7621..e3760487 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -69,7 +69,10 @@ class Trainer(TrainerIO):
         self.process_position = process_position
         self.current_gpu_name = current_gpu_name
         self.checkpoint_callback = checkpoint_callback
-        self.checkpoint_callback.save_function = self.save_checkpoint if self.checkpoint_callback is not None else None
+
+        if self.checkpoint_callback is not None:
+            self.checkpoint_callback.save_function = self.save_checkpoint
+
         self.early_stop = early_stop_callback
         self.model = None
         self.max_nb_epochs = max_nb_epochs