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

dp tests

Browse Source
This commit is contained in:
William Falcon
2019-10-18 01:06:50 +02:00
parent 5ca6b40308
commit d8920169ac
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+3 -3
View File
@@ -110,7 +110,7 @@ def test_running_test_pretrained_model_ddp():
# correct result and ok accuracy # correct result and ok accuracy
assert result == 1, 'training failed to complete' assert result == 1, 'training failed to complete'
pretrained_model = load_model(logger.experiment, save_dir, pretrained_model = load_model(logger.experiment, trainer.checkpoint_callback.filepath,
module_class=LightningTestModel) module_class=LightningTestModel)
# run test set # run test set
@@ -439,7 +439,7 @@ def test_running_test_pretrained_model_dp():
# correct result and ok accuracy # correct result and ok accuracy
assert result == 1, 'training failed to complete' assert result == 1, 'training failed to complete'
pretrained_model = load_model(logger.experiment, save_dir, pretrained_model = load_model(logger.experiment, trainer.checkpoint_callback.filepath,
module_class=LightningTestModel) module_class=LightningTestModel)
new_trainer = Trainer(**trainer_options) new_trainer = Trainer(**trainer_options)
@@ -1139,7 +1139,7 @@ def test_amp_gpu_ddp_slurm_managed():
assert trainer.resolve_root_node_address('abc[23-24, 45-40, 40]') == 'abc23' assert trainer.resolve_root_node_address('abc[23-24, 45-40, 40]') == 'abc23'
# test model loading with a map_location # test model loading with a map_location
pretrained_model = load_model(logger.experiment, save_dir) pretrained_model = load_model(logger.experiment, trainer.checkpoint_callback.filepath)
# test model preds # test model preds
[run_prediction(dataloader, pretrained_model) for dataloader in trainer.get_test_dataloaders()] [run_prediction(dataloader, pretrained_model) for dataloader in trainer.get_test_dataloaders()]