From 21914cb1c1a783fd6f09297ec02b5c3d248d7659 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 14 Jul 2019 17:15:20 -0400 Subject: [PATCH] working on single gpu init speed --- 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 71267c34..887648c1 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -82,6 +82,7 @@ class Trainer(TrainerIO): self.print_nan_grads = print_nan_grads self.data_parallel_device_ids = None self.world_size = 1 + print('-'*100) # gpus come in as a string. # if gpus = -1 then use all available devices