FullyShardedNativeNativeMixedPrecisionPlugin

class pytorch_lightning.plugins.precision.FullyShardedNativeNativeMixedPrecisionPlugin(precision, device, scaler=None)[source]

Bases: pytorch_lightning.plugins.precision.native_amp.NativeMixedPrecisionPlugin

Native AMP for Fully Sharded Native Training.

clip_grad_by_norm(*_, **__)[source]

Clip gradients by norm.

Return type:

None