yiqichen01 commited on
Commit
705d81e
·
verified ·
1 Parent(s): ce1ba0b
Files changed (2) hide show
  1. finetune/README.md +0 -1
  2. finetune/README_CN.md +2 -2
finetune/README.md CHANGED
@@ -143,7 +143,6 @@ The key parameters in the script are as follows:
143
  - `--max_steps`: Total number of training steps.
144
  - `--save_steps`: Number of steps between saving checkpoints.
145
  - `--use_lora`: Whether to use LoRA training. Also accepts `--lora_rank`, `--lora_alpha`, and `--lora_dropout` parameters. By default, LoRA is applied to "q_proj", "k_proj", "v_proj", and "o_proj". To change this, modify the code. Note: **When using LoRA training, only the LoRA weights are saved, not the base model weights.**
146
- - `--make_moe_param_leaf_module`: When using ZeRO-3 with MoE training, treat the MoE module as a leaf module, i.e., its parameters are not partitioned by ZeRO-3. This option is expected to significantly increase memory usage.
147
  - `--gradient_checkpointing`: Enable gradient checkpointing.
148
  - `--learning_rate`: Maximum learning rate during training.
149
  - `--min_lr`: Minimum learning rate during training.
 
143
  - `--max_steps`: Total number of training steps.
144
  - `--save_steps`: Number of steps between saving checkpoints.
145
  - `--use_lora`: Whether to use LoRA training. Also accepts `--lora_rank`, `--lora_alpha`, and `--lora_dropout` parameters. By default, LoRA is applied to "q_proj", "k_proj", "v_proj", and "o_proj". To change this, modify the code. Note: **When using LoRA training, only the LoRA weights are saved, not the base model weights.**
 
146
  - `--gradient_checkpointing`: Enable gradient checkpointing.
147
  - `--learning_rate`: Maximum learning rate during training.
148
  - `--min_lr`: Minimum learning rate during training.
finetune/README_CN.md CHANGED
@@ -143,8 +143,8 @@ Are you sure you want to continue connecting (yes/no)?
143
  - `--max_steps`: 训练的总步数
144
  - `--save_steps`: 每多少个 step 存储一个 checkpoint
145
  - `--use_lora`: 是否用 lora 训练,同时接收`--lora_rank`,`--lora_alpha`和`--lora_dropout`参数。lora 默认应用于 MLA(Multi-head Latent Attention)投影层:"q_a_proj", "q_b_proj", "kv_a_proj_with_mqa", "kv_b_proj", "o_proj",如果需要改变的话在代码中修改即可。注意:**使用 lora 训练时,只会保存 lora 的权重,而不会保存 base 模型的权重**
146
- - `--make_moe_param_leaf_module`:当用 zero3 以及 MoE 训练时,将 MoE 模块视作一个 leaf module,即它的参数不进行 zero3 切分,这个选项预计会显著增加显存占用
147
- - `--gradient_checkpointing`:开启梯度检查点- `--learning_rate`: 训练时的最大学习率
148
  - `--min_lr`: 训练时的最小学习率
149
  - `--use_flash_attn`: 开启 flash-attention 进行训练加速
150
 
 
143
  - `--max_steps`: 训练的总步数
144
  - `--save_steps`: 每多少个 step 存储一个 checkpoint
145
  - `--use_lora`: 是否用 lora 训练,同时接收`--lora_rank`,`--lora_alpha`和`--lora_dropout`参数。lora 默认应用于 MLA(Multi-head Latent Attention)投影层:"q_a_proj", "q_b_proj", "kv_a_proj_with_mqa", "kv_b_proj", "o_proj",如果需要改变的话在代码中修改即可。注意:**使用 lora 训练时,只会保存 lora 的权重,而不会保存 base 模型的权重**
146
+ - `--gradient_checkpointing`:开启梯度检查点
147
+ - `--learning_rate`: 训练时的最大学习率
148
  - `--min_lr`: 训练时的最小学习率
149
  - `--use_flash_attn`: 开启 flash-attention 进行训练加速
150