FullyShardedNativeMixedPrecisionPlugin

class pytorch_lightning.plugins.precision.FullyShardedNativeMixedPrecisionPlugin(*args, **kwargs)

Bases: pytorch_lightning.plugins.precision.sharded_native_amp.ShardedNativeMixedPrecisionPlugin

Native AMP for Fully Sharded Training.

clip_grad_by_norm(*_, **__)

Clip gradients by norm.

Return type:

None