FSDPMixedPrecisionPlugin¶

class lightning.pytorch.plugins.precision.FSDPMixedPrecisionPlugin(precision, device, scaler=None)[소스]¶

AMP for Fully Sharded Data Parallel (FSDP) Training.

경고

This is an experimental feature.

clip_grad_by_norm(*_, **__)[소스]¶

Clip gradients by norm.