Update FSDPPrecision to support 16-true
precision setting
#17657
Labels
fabric
lightning.fabric.Fabric
feature
Is an improvement or enhancement
refactor
strategy: fsdp
Fully Sharded Data Parallel
Milestone
Description & Motivation
We recently added
true
precision support in Fabric:#17287
#17576
The FSDPPrecision plugin needs an update to support the strings "16-true" and "bf16-true".
Pitch
Update the plugin's code by following the same pattern as in #17576.
Alternatives
No response
Additional context
No response
cc @Borda @justusschock @awaelchli @carmocca
The text was updated successfully, but these errors were encountered: