Skip to content

Commit 645db8c

Browse files
committed
Non-deprecated attr
1 parent 475bcf5 commit 645db8c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/pytorch_lightning/loops/optimization/optimizer_loop.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
_extract_hiddens,
3232
)
3333
from pytorch_lightning.plugins import ApexMixedPrecisionPlugin
34+
from pytorch_lightning.plugins.precision.native_amp import MixedPrecisionPlugin
3435
from pytorch_lightning.trainer.progress import OptimizationProgress
3536
from pytorch_lightning.utilities.exceptions import MisconfigurationException
3637
from pytorch_lightning.utilities.rank_zero import rank_zero_deprecation
@@ -366,7 +367,7 @@ def _optimizer_step(
366367
" `using_native_amp` argument. Removing this argument will avoid this message, you can expect it to"
367368
" return True."
368369
)
369-
kwargs["using_native_amp"] = self.trainer.amp_backend == "native"
370+
kwargs["using_native_amp"] = isinstance(self.trainer.precision_plugin, MixedPrecisionPlugin)
370371
self.trainer._call_lightning_module_hook(
371372
"optimizer_step",
372373
self.trainer.current_epoch,

0 commit comments

Comments
 (0)