1. api端口参数说明： src/api

-h, --help            显示帮助信息并退出

--model_name_or_path MODEL_NAME_OR_PATH

                      模型权重的路径或标识符，来自 huggingface.co/models 或 modelscope.cn/models。 (默认: None)

--adapter_name_or_path ADAPTER_NAME_OR_PATH

                      适配器权重的路径或标识符，来自 huggingface.co/models。 (默认: None)

--cache_dir CACHE_DIR

                      存储从 huggingface.co 或 modelscope.cn 下载的预训练模型的位置。 (默认: None)

--use_fast_tokenizer [USE_FAST_TOKENIZER]

                      是否使用基于 tokenizers 库支持的快速分词器。 (默认: True)

--no_use_fast_tokenizer

                      是否使用基于 tokenizers 库支持的快速分词器。 (默认: False)

--split_special_tokens [SPLIT_SPECIAL_TOKENS]

                      是否在标记化过程中拆分特殊标记。 (默认: False)

--model_revision MODEL_REVISION

                      要使用的特定模型版本（可以是分支名称、标签名称或提交 ID）。 (默认: main)

--quantization_bit QUANTIZATION_BIT

                      量化模型所需的位数。 (默认: None)

--quantization_type {fp4,nf4}

                      在 int4 训练中使用的量化数据类型。 (默认: nf4)

--double_quantization [DOUBLE_QUANTIZATION]

                      是否在 int4 训练中使用双重量化。 (默认: True)

--no_double_quantization

                      是否在 int4 训练中使用双重量化。 (默认: False)

--rope_scaling {linear,dynamic}

                      采用缩放的旋转位置嵌入。 (默认: None)

--flash_attn [FLASH_ATTN]

                      启用 FlashAttention-2 以加速训练。 (默认: False)

--shift_attn [SHIFT_ATTN]

                      启用由 LongLoRA 提出的 Shift Short Attention（S^2-Attn）。 (默认: False)

--hf_hub_token HF_HUB_TOKEN

                      用于登录 Hugging Face Hub 的身份验证令牌。 (默认: None)

--ms_hub_token MS_HUB_TOKEN

                      用于登录 ModelScope Hub 的身份验证令牌。 (默认: None)

--template TEMPLATE   用于构建训练和推断提示的模板。 (默认: None)

--dataset DATASET     要使用的提供的数据集名称。使用逗号分隔多个数据集。 (默认: None)

--dataset_dir DATASET_DIR

                      包含数据集的文件夹路径。 (默认: data)

--split SPLIT         用于训练和评估的数据集拆分。 (默认: train)

--cutoff_len CUTOFF_LEN

                      标记化后模型输入的最大长度。 (默认: 1024)

--reserved_label_len RESERVED_LABEL_LEN

                      标记化后为标签保留的最大长度。 (默认: 1)

--train_on_prompt [TRAIN_ON_PROMPT]

                      是否在提示上禁用掩码。 (默认: False)

--streaming [STREAMING]

                      启用数据集流式处理。 (默认: False)

--buffer_size BUFFER_SIZE

                      在数据集流式处理中用于随机抽样的缓冲区大小。 (默认: 16384)

--mix_strategy {concat,interleave_under,interleave_over}

                      数据集混合策略（连接/交错）（欠采样/过采样）。 (默认: concat)

--interleave_probs INTERLEAVE_PROBS

                      从数据集中抽样数据的概率。使用逗号分隔多个数据集。 (默认: None)

--overwrite_cache [OVERWRITE_CACHE]

                      覆盖缓存的训练和评估集。 (默认: False)

--preprocessing_num_workers PREPROCESSING_NUM_WORKERS

                      用于预处理的进程数。 (默认: None)

--max_samples MAX_SAMPLES

                      用于调试目的，截断每个数据集的示例数。 (默认: None)

--eval_num_beams EVAL_NUM_BEAMS

                      

 用于评估的束搜索数。该参数将传递给 `model.generate`。 (默认: None)

--ignore_pad_token_for_loss [IGNORE_PAD_TOKEN_FOR_LOSS]

                      是否在损失计算中忽略与填充标签相对应的标记。 (默认: True)

--no_ignore_pad_token_for_loss

                      是否在损失计算中忽略与填充标签相对应的标记。 (默认: False)

--val_size VAL_SIZE   开发集的大小，应为整数或范围为 `[0,1)` 的浮点数。 (默认: 0)

--sft_packing [SFT_PACKING]

                      在监督微调阶段对问题和答案进行打包。 (默认: False)

--cache_path CACHE_PATH

                      保存或加载预处理数据集的路径。 (默认: None)

--export_dir EXPORT_DIR

                      保存导出模型的目录路径。 (默认: None)

--export_size EXPORT_SIZE

                      导出模型的文件分片大小（以 GB 为单位）。 (默认: 1)

--export_quantization_bit EXPORT_QUANTIZATION_BIT

                      用于量化导出模型的位数。 (默认: None)

--export_quantization_dataset EXPORT_QUANTIZATION_DATASET

                      用于量化导出模型的数据集路径或数据集名称。 (默认: None)

--export_quantization_nsamples EXPORT_QUANTIZATION_NSAMPLES

                      用于量化的样本数。 (默认: 128)

--export_quantization_maxlen EXPORT_QUANTIZATION_MAXLEN

                      用于量化的模型输入的最大长度。 (默认: 1024)

--dpo_beta DPO_BETA   DPO 损失的 beta 参数。 (默认: 0.1)

--dpo_loss {sigmoid,hinge}

                      要使用的 DPO 损失类型。 (默认: sigmoid)

--dpo_ftx DPO_FTX     DPO 训练中的监督微调损失系数。 (默认: 0)

--ppo_buffer_size PPO_BUFFER_SIZE

                      在 PPO 优化步骤中创建经验缓冲区的小批次数。 (默认: 1)

--ppo_epochs PPO_EPOCHS

                      在 PPO 优化步骤中执行的周期数。 (默认: 4)

--ppo_logger PPO_LOGGER

                      在 PPO 训练中记录日志，使用 "wandb" 或 "tensorboard"。 (默认: None)

--ppo_score_norm [PPO_SCORE_NORM]

                      在 PPO 训练中使用分数归一化。 (默认: False)

--ppo_target PPO_TARGET

                      用于自适应 KL 控制的目标 KL 值，在 PPO 训练中。 (默认: 6.0)

--ppo_whiten_rewards [PPO_WHITEN_REWARDS]

                      在 PPO 训练中计算优势之前是否漂白奖励。 (默认: False)

--ref_model REF_MODEL

                      用于 PPO 或 DPO 训练的参考模型路径。 (默认: None)

--ref_model_adapters REF_MODEL_ADAPTERS

                      参考模型的适配器路径。 (默认: None)

--ref_model_quantization_bit REF_MODEL_QUANTIZATION_BIT

                      用于量化参考模型的位数。 (默认: None)

--reward_model REWARD_MODEL

                      用于 PPO 训练的奖励模型路径。 (默认: None)

--reward_model_adapters REWARD_MODEL_ADAPTERS

                      奖励模型的适配器路径。 (默认: None)

--reward_model_quantization_bit REWARD_MODEL_QUANTIZATION_BIT

                      用于量化奖励模型的位数。 (默认: None)

--reward_model_type {lora,full,api}

                      在 PPO 训练中使用的奖励模型类型。LoRA 模型仅支持 LoRA 训练。 (默认: lora)

--additional_target ADDITIONAL_TARGET

                      除 LoRA 层之外要设置为可训练并保存在最终检查点中的模块的名称。 (默认: None)

--lora_alpha LORA_ALPHA

                      LoRA 微调的比例因子（默认为 lora_rank * 2）。 (默认: None)

--lora_dropout LORA_DROPOUT

                      LoRA 微调的 dropout 率。 (默认: 0.1)

--lora_rank LORA_RANK

                      LoRA 微调的内在维度。 (默认: 8)

--lora_target LORA_TARGET

                      要应用 LoRA 的目标模块的名称。使用逗号分隔多个模块。 (默认: None)

--create_new_adapter [CREATE_NEW_ADAPTER]

                      是否创建具有随机初始化权重的新适配器。 (默认: False)

--name_module_trainable NAME_MODULE_TRAINABLE

                      部分参数（冻结）微调的可训练模块的名称。使用逗号分隔多个模块。 (默认: mlp)

--num_layer_trainable NUM_LAYER_TRAINABLE

                      部分参数（冻结）微调的可训练层数。 (默认: 3)

--stage {pt,sft,rm,ppo,dpo}

                      训练中将执行哪个阶段。 (默认: sft)

--finetuning_type {lora,freeze,full}

                      使用哪种微调方法。 (默认: lora)

--upcast_layernorm [UPCAST_LAYERNORM]

                      是否在 fp32 中上升 layernorm 权重。 (默认: False)

--plot_loss [PLOT_LOSS]

                      是否在微调后绘制训练损失。 (默认: False)

--do_sample [DO_SAMPLE]

                      是否使用抽样，否则使用贪婪解码。 (默认: True)

--no_do_sample        是否使用抽样，否则使用贪婪解码。 (默认: False)

--temperature TEMPERATURE

                      用于调整下一个标记概率的值。 (默认: 0.95)

--top_p TOP_P         保留的最概然令牌集，其概率加起来达到 top_p 或更高。 (默认: 0.7)

  --top_k TOP_K         保留的最高概率词汇令牌数量，用于 top-k 过滤。 (默认: 50)

  --num_beams NUM_BEAMS

                        用于波束搜索的波束数量。1 表示不进行波束搜索。 (默认: 1)

  --max_length MAX_LENGTH

                        生成令牌的最大长度。可以被 max_new_tokens 覆盖。 (默认: 512)

  --max_new_tokens MAX_NEW_TOKENS

                        生成的令牌的最大数量，忽略提示中的令牌数量。 (默认: 512)

  --repetition_penalty REPETITION_PENALTY

                        重复惩罚的参数。1.0 表示无惩罚。 (默认: 1.0)

  --length_penalty LENGTH_PENALTY

                        用于基于波束的生成的长度的指数惩罚。 (默认: 1.0)

精彩文章

评论可见，请评论后查看内容，谢谢！！！
 您阅读本篇文章共花了：