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

updated reqs

Browse Source
This commit is contained in:
William Falcon
2019-07-24 09:32:51 -04:00
parent d77914e466
commit a8a8ccb499
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+6
View File
@@ -59,6 +59,7 @@ def test_cpu_model():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,
@@ -86,6 +87,7 @@ def test_single_gpu_model():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,
@@ -118,6 +120,7 @@ def test_multi_gpu_model_dp():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,
@@ -150,6 +153,7 @@ def test_multi_gpu_model_ddp():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,
@@ -183,6 +187,7 @@ def test_amp_gpu_ddp():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,
@@ -217,6 +222,7 @@ def test_amp_gpu_dp():
model = get_model()
trainer = Trainer(
progress_bar=False,
experiment=get_exp(),
max_nb_epochs=1,
train_percent_check=0.4,