From df7b6d958ee59d7dcc5936b8b067e87fd2b67ce2 Mon Sep 17 00:00:00 2001 From: Pariente Manuel Date: Sat, 30 Nov 2019 20:50:50 +0100 Subject: [PATCH] Correct behavior for argument gpus in Trainer (#561) --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 87b0db29..fc340859 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -132,7 +132,7 @@ class Trainer(TrainerIOMixin, self.gradient_clip_val = gradient_clip_val self.check_val_every_n_epoch = check_val_every_n_epoch self.track_grad_norm = track_grad_norm - self.on_gpu = gpus is not None and torch.cuda.is_available() + self.on_gpu = True if (gpus and torch.cuda.is_available()) else False self.process_position = process_position self.weights_summary = weights_summary self.max_nb_epochs = max_nb_epochs