NativeMixedPrecisionPlugin¶
-
class
pytorch_lightning.plugins.precision.
NativeMixedPrecisionPlugin
[source]¶ Bases:
pytorch_lightning.plugins.precision.mixed.MixedPrecisionPlugin
Plugin for native mixed precision training with
torch.cuda.amp
.-
backward
(model, closure_loss, optimizer, opt_idx, should_accumulate, *args, **kwargs)[source]¶ performs the actual backpropagation
- Parameters
- Return type
-