FSDPPrecision¶
- class lightning.fabric.plugins.precision.FSDPPrecision(precision, device, scaler=None)[source]¶
Bases:
MixedPrecision
AMP for Fully Sharded Data Parallel training.
Warning
This is an experimental feature.
Bases: MixedPrecision
AMP for Fully Sharded Data Parallel training.
Warning
This is an experimental feature.