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()