added single gpu train test

Author: williamFalcon

pytorch_lightning/models/trainer.py

@@ -854,7 +854,7 @@ def __run_tng_batch(self, data_batch, batch_nb):
             for i, x in enumerate(data_batch):
                 if isinstance(x, torch.Tensor):
                     data_batch[i] = x.cuda(gpu_id)
-            output = self.model.training_step(data_batch.cuda(self.data_parallel_device_ids[0]), batch_nb)
+            output = self.model.training_step(data_batch, batch_nb)
 
         else:
             output = self.model.training_step(data_batch, batch_nb)