mymusise / ChatGLM-Tuning

基于ChatGLM-6B + LoRA的Fintune方案
MIT License
3.71k stars 444 forks source link

能不能让程序多一个 epoch的限制 #210

Open wilson9x1 opened 1 year ago

wilson9x1 commented 1 year ago

比如 --num_epochs 3

这样我跑满3轮 ,即使没有max_step 自动也结束了?

nuoma commented 1 year ago

看一下trainer的文档,直接用num_of_epochs就行,去掉max steps

Sauloo-huen commented 1 year ago

看一下trainer的文档,直接用num_of_epochs就行,去掉max steps

您好,能指路一下文档在哪看嘛?

yinzhang809 commented 1 year ago

看一下trainer的文档,直接用num_of_epochs就行,去掉max steps

您好,能指路一下文档在哪看嘛?

https://huggingface.co/docs/transformers/v4.28.1/en/main_classes/trainer#transformers.TrainingArguments

@Sauloo-huen

wilson9x1 commented 1 year ago

刚刚学完回来 可以这样--num_train_epochs 10 去掉max steps 然后应该能生效了

Sauloo-huen commented 1 year ago

非常感谢!

------------------ 原始邮件 ------------------ 发件人: "mymusise/ChatGLM-Tuning" @.>; 发送时间: 2023年5月6日(星期六) 下午3:27 @.>; @.**@.>; 主题: Re: [mymusise/ChatGLM-Tuning] 能不能让程序多一个 epoch的限制 (Issue #210)

刚刚学完回来 可以这样--num_train_epochs 10 去掉max steps 然后应该能生效了

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

Sauloo-huen commented 1 year ago

非常感谢!

------------------ 原始邮件 ------------------ 发件人: "mymusise/ChatGLM-Tuning" @.>; 发送时间: 2023年5月6日(星期六) 下午2:39 @.>; @.**@.>; 主题: Re: [mymusise/ChatGLM-Tuning] 能不能让程序多一个 epoch的限制 (Issue #210)

看一下trainer的文档,直接用num_of_epochs就行,去掉max steps

您好,能指路一下文档在哪看嘛?

https://huggingface.co/docs/transformers/v4.28.1/en/main_classes/trainer#transformers.TrainingArguments

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>