mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
ignoring dist parallel forward
This commit is contained in:
1 parent
9be15aa29f
commit
4260769e14
1 file changed
+1
-1
@@ -77,7 +77,7 @@ class LightningDistributedDataParallel(DistributedDataParallel):
|
||||
def parallel_apply(self, replicas, inputs, kwargs):
|
||||
return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
|
||||
|
||||
def forward(self, *inputs, **kwargs):
|
||||
def forward(self, *inputs, **kwargs): # pragma: no cover
|
||||
self._sync_params()
|
||||
if self.device_ids:
|
||||
inputs, kwargs = self.scatter(inputs, kwargs, self.device_ids)
|
||||
|
||||
Reference in new issue
Block a user