foundation-model-stack / fms-hf-tuning

🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.
Apache License 2.0
9 stars 30 forks source link

feat: Pass trainer controller metrics to operations #175

Open seshapad opened 2 weeks ago

seshapad commented 2 weeks ago

Is your feature request related to a problem? Please describe.

Trainer control operations have no access to trainer control metrics.

Describe the solution you'd like

Trainer controller metrics should be passed to the operations to allow for their usage in performing operating tasks such as logging.

Describe alternatives you've considered

NA

Additional context

NA