LLaMA-Factory icon indicating copy to clipboard operation
LLaMA-Factory copied to clipboard

Published 20 hours ago • verified publisher icon hiyouga

wandb 初始化问题

Open thkelper opened this issue 9 months ago • 1 comments

Reminder

  • [X] I have read the README and searched the existing issues.

我想要用wandb初始化name，project以及entity。 但是我没有找到相应的接口，只找到一个report_to wandb的参数，但是这个设置无法满足我输入定制参数的需求。 如果我在train_bash.py中自己写一个wandb.init，那么在多进程的情况下，有多少进程就会增加多少条wandb记录。

thkelper avatar May 01 '24 08:05 thkelper

涉及多进程的东西用这个判断一下 https://github.com/hiyouga/LLaMA-Factory/blob/d27e6a46b4aad61c494dd22eed138306366750ee/src/llmtuner/train/sft/workflow.py#L78

hiyouga avatar May 01 '24 08:05 hiyouga

plot_loss

所以大佬你是咋修改的呢

CXLiang123 avatar Jul 23 '24 02:07 CXLiang123

@.***，这个if语句下面增加代码即可。

发件人: CXLiang @.> 日期: 星期二, 2024年7月23日 10:49 收件人: hiyouga/LLaMA-Factory @.> 抄送: kelper Young @.>, Author @.> 主题: Re: [hiyouga/LLaMA-Factory] wandb 初始化问题 (Issue #3537)

plot_loss

所以大佬你是咋修改的呢

― Reply to this email directly, view it on GitHubhttps://github.com/hiyouga/LLaMA-Factory/issues/3537#issuecomment-2244148644, or unsubscribehttps://github.com/notifications/unsubscribe-auth/ALGYVZSRNWPBA5KSIZLF3NTZNXACXAVCNFSM6AAAAABHBUYBK2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDENBUGE2DQNRUGQ. You are receiving this because you authored the thread.Message ID: @.***>

thkelper avatar Jul 23 '24 15:07 thkelper