added cpu model test

pytorch_lightning/examples/new_project_templates/lightning_module_template.py

@@ -108,7 +108,7 @@ class LightningTemplateModel(LightningModule):
 
         output = OrderedDict({
             'val_loss': loss_val,
-            'val_acc': torch.tensor(val_acc).cuda(loss_val.device.index),
+            'val_acc': torch.tensor(val_acc).type(loss_val.dtype),
         })
 
         # can also return just a scalar instead of a dict (return loss_val)

pytorch_lightning/models/trainer.py

@@ -436,6 +436,10 @@ class Trainer(TrainerIO):
 
             self.__run_pretrain_routine(model)
 
+        # return 1 when finished
+        # used for testing or when we need to know that training succeeded
+        return 1
+
     def dp_train(self, model):
 
         # CHOOSE OPTIMIZER