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

refactored model tests

Browse Source
This commit is contained in:
William Falcon
2019-07-24 12:04:48 -04:00
parent de95179556
commit e5f73304b3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+4 -4
View File
@@ -15,9 +15,9 @@ torch.manual_seed(SEED)
np.random.seed(SEED)
# -----------------
# ------------------------------------------------------------------------
# TESTS
# -----------------
# ------------------------------------------------------------------------
def test_cpu_model():
"""
Make sure model trains on CPU
@@ -222,9 +222,9 @@ def test_amp_gpu_ddp():
clear_save_dir()
# -----------------
# ------------------------------------------------------------------------
# UTILS
# -----------------
# ------------------------------------------------------------------------
def get_model():
# set up model with these hyperparams
root_dir = os.path.dirname(os.path.realpath(__file__))