Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Sourab Mangrulkar's avatar
    move fsdp handling to accelerate (#23158) · 0b774074
    Sourab Mangrulkar authored 
    * mixed precision support via accelerate

* fix issues

* fix for the sharded ddp case

* fix flax and tf failing tests

* `refactor the place to create `Accelerator` object

* move ddp prep to accelerate

* fix 馃槄

* resolving comments

* move fsdp handling to accelerate

* fixex

* fix saving
    0b774074
trainer.py 187 KB