added clean slurm save load test

This commit is contained in:
William Falcon 2019-07-26 22:39:44 -04:00
parent 2a4081e537
commit 3224365190
1 changed files with 2 additions and 1 deletions

View File

@ -74,8 +74,9 @@ def test_cpu_slurm_save_load():
saved_filepath = trainer.hpc_save(save_dir, exp)
assert os.path.exists(saved_filepath)
# wipe-out trainer model
# wipe-out trainer and model
# we want to see if the weights come back correctly
trainer = Trainer(**trainer_options)
trainer.model = LightningTestModel(hparams)
# test HPC loading