From 2e30dd94bc9fc1464240deebfde0a085b334c8d6 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 11:35:46 -0400
Subject: [PATCH] added safeguards for callbacks in loading saving

---
 tests/debug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/debug.py b/tests/debug.py
index a9700356..7f97d86d 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -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: