Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Removed lines

This commit is contained in:
SeanNaren 2020-11-25 21:18:11 +00:00
parent 321e63ae8b
commit 28afc46be2
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/overrides/fairscale.py
View File

@ -16,7 +16,6 @@ from pytorch_lightning.utilities import _module_available, NATIVE_AMP_AVALAIBLE
if _module_available('fairscale.nn.data_parallel.sharded_ddp') and NATIVE_AMP_AVALAIBLE:
from fairscale.nn.data_parallel.sharded_ddp import ShardedDataParallel
class LightningShardedDataParallel(ShardedDataParallel):
def forward(self, *inputs, **kwargs):
@ -31,7 +30,6 @@ if _module_available('fairscale.nn.data_parallel.sharded_ddp') and NATIVE_AMP_AV
outputs = self.module.validation_step(*inputs, **kwargs)
return outputs
FAIRSCALE_SHARDED_AVAILABLE = True
else:
FAIRSCALE_SHARDED_AVAILABLE = False