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

updated reqs

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 09:32:51 -04:00
1 parent d77914e466
commit a8a8ccb499
1 file changed
+6
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,