optimum
5017d066 - Refactoring FSDP. (#1586)

Commit

1 year ago

Refactoring FSDP. (#1586) * refactor fsdp * add trainer * remove hidden layers * update dockerfile --------- Co-authored-by: Adam Louly <adamlouly@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net> Co-authored-by: JingyaHuang <huang_jingya@outlook.com>

References

#1586 - Refactoring FSDP.

Author

AdamLouly

Parents

1a807fc4

Files3

examples/onnxruntime/training/docker
- Dockerfile-ort1.16.3-cu118
optimum/onnxruntime
- trainer.py
tests/onnxruntime/docker
- Dockerfile_onnxruntime_trainer

optimum 5017d066 - Refactoring FSDP. (#1586)

optimum
5017d066 - Refactoring FSDP. (#1586)