DeepSpeed
2518cc42 - Support `exclude_frozen_parameters` for `zero_to_fp32.py` script (#4979)

Commit
1 year ago
Support `exclude_frozen_parameters` for `zero_to_fp32.py` script (#4979) Adds support for the `zero_to_fp32.py` script to merge only the trainable parameters through a new argument `only_trainable_params`. Fixes #3437 --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Author
Parents
Loading