Shortcuts

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.