mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
added clean slurm save load test
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user