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:15:26 -04:00
parent e5c92e75ec
commit 8d44ebbb38
1 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+25 -2
View File
@@ -36,6 +36,13 @@ def get_exp():
return exp
def clear_tt_dir():
root_dir = os.path.dirname(os.path.realpath(__file__))
tt_dir = os.path.join(root_dir, 'tests_tt_dir')
if os.path.exists(tt_dir):
os.remove(tt_dir)
def assert_ok_acc(trainer):
# this model should get 0.80+ acc
assert trainer.tng_tqdm_dic['val_acc'] > 0.80, "model failed to get expected 0.80 validation accuracy"
@@ -54,12 +61,14 @@ def test_cpu_model():
train_percent_check=0.4,
val_percent_check=0.4
)
result = trainer.fit(model)
assert result == 1, 'cpu model failed to complete'
# correct result and ok accuracy
assert result == 1, 'cpu model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
def test_single_gpu_model():
"""
@@ -82,9 +91,12 @@ def test_single_gpu_model():
result = trainer.fit(model)
# correct result and ok accuracy
assert result == 1, 'single gpu model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
def test_multi_gpu_model_dp():
"""
@@ -110,9 +122,12 @@ def test_multi_gpu_model_dp():
result = trainer.fit(model)
# correct result and ok accuracy
assert result == 1, 'multi-gpu dp model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
def test_multi_gpu_model_ddp():
"""
@@ -139,9 +154,12 @@ def test_multi_gpu_model_ddp():
result = trainer.fit(model)
# correct result and ok accuracy
assert result == 1, 'multi-gpu ddp model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
def test_amp_gpu_ddp():
"""
@@ -169,9 +187,12 @@ def test_amp_gpu_ddp():
result = trainer.fit(model)
# correct result and ok accuracy
assert result == 1, 'amp + ddp model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
def test_amp_gpu_dp():
"""
@@ -199,9 +220,11 @@ def test_amp_gpu_dp():
result = trainer.fit(model)
# correct result and ok accuracy
assert result == 1, 'amp + gpu model failed to complete'
assert_ok_acc(trainer)
clear_tt_dir()
if __name__ == '__main__':
pytest.main([__file__])