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

pkg relative imports

Browse Source 
* split requirements.txt
* pytest verbose
This commit is contained in:
Jiri BOROVEC
2019-08-05 10:52:09 +02:00
parent 92f8c57ff5
commit 469941a528
20 changed files with 54 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/testing/lm_test_module.py
+10 -9
View File
@@ -1,17 +1,18 @@
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 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 pytorch_lightning.root_module.root_module import LightningModule
import pytorch_lightning as ptl
from ..root_module.root_module import LightningModule
from pytorch_lightning import data_loader
class LightningTestModel(LightningModule):
@@ -214,15 +215,15 @@ class LightningTestModel(LightningModule):
return loader
@ptl.data_loader
@data_loader
def tng_dataloader(self):
return self.__dataloader(train=True)
@ptl.data_loader
@data_loader
def val_dataloader(self):
return self.__dataloader(train=False)
@ptl.data_loader
@data_loader
def test_dataloader(self):
return self.__dataloader(train=False)