Update deprecation message

Lightning-AI · Jul 28, 2021 · 2f20b65 · 2f20b65
1 parent 16299b8
commit 2f20b65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1964,7 +1964,7 @@ def model_size(self) -> float:
         """
         rank_zero_deprecation(
             "The `LightningModule.model_size` property was deprecated in v1.5 and will be removed in v1.7. Please "
-            "use the get_model_size_mb method under utilities/model_helpers.py"
+            "use the get_model_size_mb method under utilities/memory.py"
         )
 
         tmp_name = f"{uuid.uuid4().hex}.pt"