search

mosaicml / llm-foundry

LLM training code for Databricks foundation models
https://www.databricks.com/blog/introducing-dbrx-new-state-art-open-llm
Apache License 2.0
3.83k stars 502 forks source link

Small refactor for update batch size #1293

Closed dakinggg closed 1 week ago

dakinggg commented 1 week ago

Just a small cleanup of the batch size update function