FSDPPrecision¶
- class lightning.fabric.plugins.precision.FSDPPrecision(precision, device, scaler=None)[source]¶
Bases:
MixedPrecisionAMP for Fully Sharded Data Parallel training.
Warning
This is an experimental feature.
- Docs >
- lightning.fabric.plugins.precision >
- FSDPPrecision
-
Shortcuts
To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.
