FSDPMixedPrecisionPlugin¶
- class lightning.pytorch.plugins.precision.FSDPMixedPrecisionPlugin(precision, device, scaler=None)[source]¶
Bases:
lightning.pytorch.plugins.precision.amp.MixedPrecisionPlugin
AMP for Fully Sharded Data Parallel (FSDP) Training.
Warning
This is an experimental feature.