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

View File

@ -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