FSDP and Gradient Clipping #15047
Unanswered
PedroRASB
asked this question in
DDP / multi-GPU / multi-node
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hello,
I am trying to use gradient clipping (setting gradient_clip_val in the lightning Trainer) with fsdp, but it returns an error: "pytorch_lightning.utilities.exceptions.MisconfigurationException:
gradient_clip_algorithm='norm'
is currently not supported forFullyShardedNativeMixedPrecisionPlugin
".Are there plans to implement gradient clipping with fsdp in Pytorch Lightning?
Thank you.
Beta Was this translation helpful? Give feedback.
All reactions