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

Remove else check

This commit is contained in:
SeanNaren 2020-11-26 18:42:06 +00:00
parent c0e148bc27
commit 2e50c2e653
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_lightning/overrides/fairscale.py
View File

@ -13,6 +13,7 @@
# limitations under the License. # limitations under the License.
from pytorch_lightning.utilities import FAIRSCALE_AVAILABLE from pytorch_lightning.utilities import FAIRSCALE_AVAILABLE
LightningShardedDataParallel = None
if FAIRSCALE_AVAILABLE: if FAIRSCALE_AVAILABLE:
from fairscale.nn.data_parallel.sharded_ddp import ShardedDataParallel from fairscale.nn.data_parallel.sharded_ddp import ShardedDataParallel
@ -29,6 +30,3 @@ if FAIRSCALE_AVAILABLE:
else: else:
outputs = self.module.validation_step(*inputs, **kwargs) outputs = self.module.validation_step(*inputs, **kwargs)
return outputs return outputs
else:
LightningShardedDataParallel = None