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

update by flake8

Browse Source
This commit is contained in:
Jiri BOROVEC
2019-08-06 22:45:46 +02:00
parent 4e0b9c50e7
commit d9bfe964f9
15 changed files with 226 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/debug.py
+2 -1
View File
@@ -107,7 +107,8 @@ def load_model(exp, save_dir):
checkpoints = [x for x in os.listdir(save_dir) if '.ckpt' in x]
weights_dir = os.path.join(save_dir, checkpoints[0])
trained_model = LightningTemplateModel.load_from_metrics(weights_path=weights_dir, tags_csv=tags_path, on_gpu=True)
trained_model = LightningTemplateModel.load_from_metrics(weights_path=weights_dir,
tags_csv=tags_path, on_gpu=True)
assert trained_model is not None, 'loading model failed'