majic31
2024-12-24 23e7ddebccd3b05cf7ef89809bcfe565ad6dfa1f
examples/industrial_data_pretraining/seaco_paraformer/finetune.sh
@@ -50,7 +50,7 @@
    --nproc_per_node $gpu_num \
    --node_rank ${RANK:-0} \
    --master_addr ${MASTER_ADDR:-127.0.0.1} \
    --master_port ${MASTER_PORT: 26669}
    --master_port ${MASTER_PORT:-26669}
"
echo $DISTRIBUTED_ARGS
@@ -78,5 +78,6 @@
++train_conf.avg_nbest_model=10 \
++train_conf.use_deepspeed=false \
++train_conf.deepspeed_config=${deepspeed_config} \
++train_conf.find_unused_parameters=true \
++optim_conf.lr=0.0002 \
++output_dir="${output_dir}" &> ${log_file}