Skip to content

Commit e305149

Browse files
committed
added single gpu train test
1 parent c7e8436 commit e305149

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/models/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def __init__(self,
196196
warnings.warn(w)
197197

198198
# remove dp and ddp when requesting single gpu
199-
if len(self.data_parallel_device_ids) == 1:
199+
if self.data_parallel_device_ids is not None and len(self.data_parallel_device_ids) == 1:
200200
self.use_ddp = False
201201
self.use_dp = False
202202
self.single_gpu = True

0 commit comments

Comments
 (0)