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