feat: Return entire log-line in loss.py metric - Githubissues

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: Return entire log-line in loss.py metric #176

Open seshapad opened 2 weeks ago

seshapad commented 2 weeks ago

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

Loss metric does not have information on the step and epoch it was generated.

Describe the solution you'd like

Return entire log-line in loss.py metric

Describe alternatives you've considered

NA

Additional context

NA