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

made root note address individually testable

Browse Source
This commit is contained in:
William Falcon
2019-07-24 18:55:38 -04:00
parent ccd4018dd9
commit 750fefac0c
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
@@ -68,6 +68,12 @@ def test_amp_gpu_ddp_slurm_managed():
# correct result and ok accuracy
assert result == 1, 'amp + ddp model failed to complete'
# test root model address
assert trainer.resolve_root_node_address('abc') == 'abc'
assert trainer.resolve_root_node_address('abc[23]') == 'abc23'
assert trainer.resolve_root_node_address('abc[23-24]') == 'abc24'
assert trainer.resolve_root_node_address('abc[23-24, 45-40, 40]') == 'abc23'
# test model loading
pretrained_model = load_model(exp, save_dir, True)