pytorch-lightning
Rudimentary support for weight averaging (EMA) with FSDP
#21414
Open
Overview
Commits
1
Changes
View On GitHub
Commits
  • Summon all model parameters before updating the average model, when using FSDP
    Seppo Enarvi committed 6 days ago
FAQ Terms Privacy Refunds Impressum
Loading