shibing624 / MedicalGPT

MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training Pipeline. 训练医疗大模型,实现了包括增量预训练(PT)、有监督微调(SFT)、RLHF、DPO、ORPO。
Apache License 2.0
3.24k stars 492 forks source link

num_train_epochs这个参数好像只在训练的.sh文件中出现,在代码中没有出现过? #287

Closed erichuazhou closed 9 months ago

erichuazhou commented 9 months ago

Describe the bug

Please provide a clear and concise description of what the bug is. If applicable, add screenshots to help explain your problem, especially for visualization related problems.

num_train_epochs本意是控制训练的epoch数量,但是num_train_epochs这个参数好像只在训练的.sh文件中出现,在代码中搜索不到? @shibing624

shibing624 commented 9 months ago

执行不会报错的。看下transformers的Trainer源码。