mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
added single node distdataparallel
This commit is contained in:
1 parent
32eddf492e
commit
23137ea08a
1 file changed
+1
@@ -28,6 +28,7 @@ class LightningDataParallel(DataParallel):
|
||||
"""
|
||||
|
||||
def parallel_apply(self, replicas, inputs, kwargs):
|
||||
print('parallel_apply called')
|
||||
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user