AI-Hypercomputer / AI-Hypercomputer/maxtext

wandb logging support

Aperta
#2,434 5 commenti 0 reazioni 1 assegnatario Rivendicata da @dipannita08 Vedi su GitHub
feature request
Lingua principale
Python
Stelle
2.4k
Fork
607
Merge medio
2g 19h
PR unite (30g)
158

Descrizione

### 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_

Guida per i contributori

Apri la guida per i contributori

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.