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

Fix import order

This commit is contained in:
SeanNaren 2020-11-26 17:07:48 +00:00
parent 8a0c8fe0bd
commit 29e310807c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/test_sharded_correctness.py
View File

@ -9,7 +9,7 @@ from torch.utils.data.distributed import DistributedSampler
from pytorch_lightning import Trainer, seed_everything from pytorch_lightning import Trainer, seed_everything
from pytorch_lightning.plugins.sharded_plugin import DDPShardedPlugin from pytorch_lightning.plugins.sharded_plugin import DDPShardedPlugin
from pytorch_lightning.utilities import NATIVE_AMP_AVALAIBLE, FAIRSCALE_AVAILABLE from pytorch_lightning.utilities import FAIRSCALE_AVAILABLE, NATIVE_AMP_AVALAIBLE
from tests.base.boring_model import BoringModel, RandomDataset from tests.base.boring_model import BoringModel, RandomDataset