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

update codecov

Browse Source
This commit is contained in:
Jiri BOROVEC
2019-08-07 14:32:32 +02:00
parent 421c4fab7d
commit 86a90bfefd
5 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+6 -4
View File
@@ -1,20 +1,22 @@
import os
import shutil
import warnings
from argparse import Namespace
import pytest
import numpy as np
import torch
from pytorch_lightning import Trainer
from examples import LightningTemplateModel
from pytorch_lightning.testing.lm_test_module import LightningTestModel
from argparse import Namespace
from test_tube import Experiment, SlurmCluster
# sys.path += [os.path.abspath('..'), os.path.abspath('../..')]
from pytorch_lightning import Trainer
from pytorch_lightning.testing.lm_test_module import LightningTestModel
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
from pytorch_lightning.utilities.debugging import MisconfigurationException
from pytorch_lightning.root_module import memory
from pytorch_lightning.models.trainer import reduce_distributed_output
from pytorch_lightning.root_module import model_saving
from examples import LightningTemplateModel
SEED = 2334
torch.manual_seed(SEED)