wip

Lightning-AI · Jun 11, 2020 · 9b2e777 · 9b2e777
1 parent 3be78f3
commit 9b2e777
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pytorch_lightning/trainer/auto_mix_precision.py b/pytorch_lightning/trainer/auto_mix_precision.py
@@ -20,6 +20,11 @@ class TrainerAMPMixin(ABC):
     use_native_amp: bool
 
     def init_amp(self, use_amp):
+        # TODO: remove in v 0.9.0
+        if self.use_native_amp:
+            rank_zero_warn("`amp_level` has been deprecated since v0.7.4 (native amp does not require it)"
+                           " and this argument will be removed in v0.9.0", DeprecationWarning)
+
         # Backward compatibility, TODO: remove in v0.9.0
         if use_amp is not None:
             rank_zero_warn("`use_amp` has been replaced by `precision` since v0.7.0"