AI-Hypercomputer / AI-Hypercomputer/maxtext

wandb logging support

未關閉
#2,434 5 則留言 0 個 reaction 已指派 1 人 已被 @dipannita08 認領 在 GitHub 檢視
feature request
主要語言
Python
星號
2.4k
分支
607
平均合併
2 天 19 小時
30 天內合併 PR
158

描述

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

貢獻指南

開啟貢獻指南

評估

這個 Issue 還沒有評估資料。

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。