mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-30 13:40:46 +08:00
set dp as default backend
This commit is contained in:
1 parent
81d39786d9
commit
4db32984c6
1 file changed
-2
@@ -67,7 +67,6 @@ class LightningDataParallel(DataParallel):
|
||||
|
||||
|
||||
def parallel_apply(self, replicas, inputs, kwargs):
|
||||
print('LDP')
|
||||
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
|
||||
|
||||
|
||||
@@ -80,7 +79,6 @@ class LightningDistributedDataParallel(DistributedDataParallel):
|
||||
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
|
||||
|
||||
def forward(self, *inputs, **kwargs):
|
||||
print('LDDP')
|
||||
self._sync_params()
|
||||
if self.device_ids:
|
||||
inputs, kwargs = self.scatter(inputs, kwargs, self.device_ids)
|
||||
|
||||
Reference in new issue
Block a user