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 摘要。