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

fixed hpc save, load. cleaned apu

Browse Source
This commit is contained in:
William Falcon
2019-07-26 22:13:06 -04:00
parent 348223a702
commit a6ae97ac09
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -1
View File
@@ -115,7 +115,7 @@ def test_model_saving_loading():
# make prediction
# assert that both predictions are the same
new_pred = model_2(x)
assert torch.eq(pred_before_saving, new_pred)
assert torch.all(torch.eq(pred_before_saving, new_pred)).item() == 1
clear_save_dir()