wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-05 16:41:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

added safeguards for callbacks in loading saving

Browse Source
This commit is contained in:
William Falcon
2019-07-24 11:35:46 -04:00
parent 55a33edd0a
commit 2e30dd94bc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/debug.py
+1 -1
View File
@@ -87,7 +87,7 @@ def main():
pdb.set_trace()
tags_path = exp.get_data_path(exp.name, exp.version)
tags_path = os.path.join(tags_path, 'meta_tags.csv')
trained_model = LightningTemplateModel.load_from_metrics(weights_path=save_dir, tags_csv=tags_path)
trained_model = LightningTemplateModel.load_from_metrics(weights_path=save_dir, tags_csv=tags_path, on_gpu=True)
# run prediction
for batch in model.test_dataloader: