AI-Hypercomputer / AI-Hypercomputer/maxtext
wandb logging support
Aberta
feature request
- Linguagem predominante
- Python
- Estrelas
- 2.4k
- Forks
- 607
- Merge médio
- 2d 19h
- PRs com merge (30d)
- 158
Descrição
### Feature or Model Request
HI MaxText developers! Recently I'm using MaxText to train LLMs, and in my fork I noticed that there's a way to patch the metric_logger to enable wandb logging for MaxText.
I wonder if this is a desired feature in MaxText. If yes, I'd love to create a PR for this since I have a working implementation.
### Additional Context
_No response_
Guia de contribuição
Avaliação
Esta issue ainda não foi avaliada.