From 904935cf981604292942e2b96c1eb4d2860c941d Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Sun, 14 Jul 2019 17:11:52 -0400
Subject: [PATCH] working on single gpu init speed

---
 pytorch_lightning/models/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 4369dcf3..71267c34 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -306,6 +306,7 @@ class Trainer(TrainerIO):
             self.single_gpu_train(model)
 
         else:
+            print('NO GPU')
             # CHOOSE OPTIMIZER
             # filter out the weights that were done on gpu so we can load on good old cpus
             self.optimizers = model.configure_optimizers()