fixed error

Lightning-AI · Oct 25, 2020 · c53aedb · c53aedb
1 parent bd2c4bf
commit c53aedb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -314,7 +314,7 @@ def test_resume_training_on_cpu():
     model_path = trainer.checkpoint_callback.best_model_path
 
     # Verify saved Tensors are on CPU
-    ckpt = torch.load(model)
+    ckpt = torch.load(model_path)
     weight_tensor = list(ckpt["state_dict"].values())[0]
     assert weight_tensor.device == torch.device("cpu")