wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Minor imports cleaning (#402)

Browse Source 
* code cleaning

* drop unused imports

* optimize imports
This commit is contained in:
Jirka Borovec
2019-10-22 11:32:40 +03:00
committed by William Falcon
parent e6e325c853
commit f18aee30a5
32 changed files with 94 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/testing/lm_test_module.py
-14
View File
@@ -1,18 +1,4 @@
import os
from collections import OrderedDict
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import optim
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
from torchvision.datasets import MNIST
from torchvision import transforms
from test_tube import HyperOptArgumentParser
from pytorch_lightning.root_module.root_module import LightningModule
from pytorch_lightning import data_loader
from .lm_test_module_base import LightningTestModelBase
from .lm_test_module_mixins import LightningValidationMixin, LightningTestMixin
pytorch_lightning/testing/lm_test_module_base.py
+3 -3
View File
@@ -4,15 +4,15 @@ from collections import OrderedDict
import torch
import torch.nn as nn
import torch.nn.functional as F
from test_tube import HyperOptArgumentParser
from torch import optim
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
from torchvision.datasets import MNIST
from torchvision import transforms
from test_tube import HyperOptArgumentParser
from torchvision.datasets import MNIST
from pytorch_lightning.root_module.root_module import LightningModule
from pytorch_lightning import data_loader
from pytorch_lightning.root_module.root_module import LightningModule
class LightningTestModelBase(LightningModule):
pytorch_lightning/testing/lm_test_module_mixins.py
-10
View File
@@ -1,17 +1,7 @@
import os
from collections import OrderedDict
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import optim
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
from torchvision.datasets import MNIST
from torchvision import transforms
from test_tube import HyperOptArgumentParser
from pytorch_lightning.root_module.root_module import LightningModule
from pytorch_lightning import data_loader