Skip to content
Snippets Groups Projects
Unverified Commit ef946404 authored by Alex Yang's avatar Alex Yang Committed by GitHub
Browse files

[Feat] Support FSDP Training (#304) 

* [Feat] Support FSDP Training

* fix version comparison

* change param format and move `FSDP_WRAP_POLICY` to wrapper file

* add docstring and type hint,reformat code

* fix type hint

* fix typo, reformat code
parent e76517c6
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Datenschutz und Impressum