accelerate
a7d6f28f - feat: add ignored_params support for fsdp2 (#3731)

Commit
152 days ago
feat: add ignored_params support for fsdp2 (#3731) * feat: add ignored_params support for fsdp2 Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * feat: add ignored_params support for fsdp2 Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * feat: add ignored_params support for fsdp2 Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * feat: add ignored_params support for fsdp2 Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * test: update testcase for fsdp2 ignored_params Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * fix: add defensive use of ignored params Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> * fix: styling errors Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com> --------- Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
Author
Parents
Loading