mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
added single gpu train test
This commit is contained in:
1 parent
c7e8436083
commit
e30514922b
1 file changed
+1
-1
@@ -196,7 +196,7 @@ class Trainer(TrainerIO):
|
||||
warnings.warn(w)
|
||||
|
||||
# remove dp and ddp when requesting single gpu
|
||||
if len(self.data_parallel_device_ids) == 1:
|
||||
if self.data_parallel_device_ids is not None and len(self.data_parallel_device_ids) == 1:
|
||||
self.use_ddp = False
|
||||
self.use_dp = False
|
||||
self.single_gpu = True
|
||||
|
||||
Reference in new issue
Block a user