ShardedNativeMixedPrecisionPlugin¶
- class pytorch_lightning.plugins.precision.ShardedNativeMixedPrecisionPlugin(precision, device, scaler=None)[source]¶
Bases:
pytorch_lightning.plugins.precision.native_amp.NativeMixedPrecisionPlugin
Native AMP for Sharded Training.