FSDPPrecision¶
- class lightning_fabric.plugins.precision.FSDPPrecision(precision, device, scaler=None)[source]¶
Bases:
lightning_fabric.plugins.precision.native_amp.MixedPrecision
AMP for Fully Sharded Data Parallel training.
Bases: lightning_fabric.plugins.precision.native_amp.MixedPrecision
AMP for Fully Sharded Data Parallel training.