rx0206 commited on
Commit
589ba77
·
verified ·
1 Parent(s): 53a84c2

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 参数名 解释
2
+ data_path 数据路径或 HuggingFace 仓库名
3
+ max_length 单条数据最大 Token 数,超过则截断
4
+ pack_to_max_length 是否将多条短数据拼接到 max_length,提高 GPU 利用率
5
+ accumulative_counts 梯度累积,每多少次 backward 更新一次参数
6
+ sequence_parallel_size 并行序列处理的大小,用于模型训练时的序列并行
7
+ batch_size 每个设备上的批量大小
8
+ dataloader_num_workers 数据加载器中工作进程的数量
9
+ max_epochs 训练的最大轮数
10
+ optim_type 优化器类型,例如 AdamW
11
+ lr 学习率
12
+ betas 优化器中的 beta 参数,控制动量和平方梯度的移动平均
13
+ weight_decay 权重衰减系数,用于正则化和避免过拟合
14
+ max_norm 梯度裁剪的最大范数,用于防止梯度爆炸
15
+ warmup_ratio 预热的比例,学习率在这个比例的训练过程中线性增加到初始学习率
16
+ save_steps 保存模型的步数间隔
17
+ save_total_limit 保存的模型总数限制,超过限制时删除旧的模型文件
18
+ prompt_template 模板提示,用于定义生成文本的格式或结构
19
+ ...... ......
20
+ 如果想充分利用显卡资源,可以将 max_length 和 batch_size 这两个参数调大。 ⚠但需要注意的是,在训练 chat 模型时调节参数 batch_size 有可能会影响对话模型的效果。