We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b912377 commit 600280bCopy full SHA for 600280b
tests/tests_pytorch/plugins/precision/test_amp.py
@@ -18,7 +18,7 @@
18
from torch import nn
19
from torch.optim import Optimizer
20
21
-from lightning.pytorch.plugins.precision import MixedPrecision
+from lightning.pytorch.plugins import MixedPrecision
22
from lightning.pytorch.utilities import GradClipAlgorithmType
23
24
0 commit comments