AI开发平台MODELARTS-yaml配置文件参数配置说明:sft_yaml样例模板

时间：2024-09-24 10:50:21

AI开发平台MODELARTS

sft_yaml样例模板

### model
model_name_or_path: /home/ma-user/ws/tokenizers/Qwen2-72B
### method
stage: sft
do_train: true
finetuning_type: full
deepspeed: examples/deepspeed/ds_z3_config.json
### dataset
dataset: identity,alpaca_en_demo
template: qwen
cutoff_len: 4096
packing: true
max_samples: 1000
overwrite_cache: true
preprocessing_num_workers: 16
### output
output_dir: /home/ma-user/ws/tokenizers/Qwen2-72B/sft
logging_steps: 2
save_steps: 5000
plot_loss: true
overwrite_output_dir: true
### train
per_device_train_batch_size: 1
gradient_accumulation_steps: 8
learning_rate: 1.0e-5
num_train_epochs: 10.0
lr_scheduler_type: cosine
warmup_ratio: 0.1
fp16: true
ddp_timeout: 180000000
include_tokens_per_second: true
include_num_input_tokens_seen: true

上一篇：AI开发平台MODELARTS-yaml配置文件参数配置说明:ds_z1_config.json样例模板

下一篇：AI开发平台MODELARTS-yaml配置文件参数配置说明:ds_z1_config.json样例模板