From 2e50c2e653ba417a80dd3517895be5dd6d06ba82 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Thu, 26 Nov 2020 18:42:06 +0000 Subject: [PATCH] Remove else check --- pytorch_lightning/overrides/fairscale.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pytorch_lightning/overrides/fairscale.py b/pytorch_lightning/overrides/fairscale.py index df49199e2a..c1d5769df6 100644 --- a/pytorch_lightning/overrides/fairscale.py +++ b/pytorch_lightning/overrides/fairscale.py @@ -13,6 +13,7 @@ # limitations under the License. from pytorch_lightning.utilities import FAIRSCALE_AVAILABLE +LightningShardedDataParallel = None if FAIRSCALE_AVAILABLE: from fairscale.nn.data_parallel.sharded_ddp import ShardedDataParallel @@ -29,6 +30,3 @@ if FAIRSCALE_AVAILABLE: else: outputs = self.module.validation_step(*inputs, **kwargs) return outputs - -else: - LightningShardedDataParallel = None