From baa139f97a5472e1ce3fc4a4234e3896e6c16faa Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 18 Jul 2019 11:09:00 -0400
Subject: [PATCH] added training router

---
 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 b9000ff4..acff43fc 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -319,6 +319,7 @@ class Trainer(TrainerIO):
         elif self.use_dp:
             self.dp_train(model)
 
+        # ON CPU
         else:
             # CHOOSE OPTIMIZER
             # filter out the weights that were done on gpu so we can load on good old cpus