AI-Hypercomputer / AI-Hypercomputer/maxtext
wandb logging support
Ouverte
feature request
- Langage dominant
- Python
- Étoiles
- 2.4k
- Forks
- 607
- Merge moyen
- 2 j 19 h
- PR mergées (30 j)
- 158
Description
### 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_
Guide de contribution
Ouvrir le guide de contribution
Évaluation
Cette issue n'a pas encore été évaluée.