wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +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
pl_examples/basic_examples/lightning_module_template.py
+7 -6
View File
@@ -2,16 +2,17 @@
Example template for defining a system
"""
import os
from collections import OrderedDict
import torch.nn as nn
from torchvision.datasets import MNIST
import torchvision.transforms as transforms
import torch
import torch.nn.functional as F
from argparse import ArgumentParser
from collections import OrderedDict
import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision.transforms as transforms
from torch import optim
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
from torchvision.datasets import MNIST
import pytorch_lightning as pl
from pytorch_lightning.root_module.root_module import LightningModule