| | |
|---|
| | | # model_name from model_hub, or model_dir in local path |
|---|
| | | |
|---|
| | | ## option 1, download model automatically |
|---|
| | | model_name_or_model_dir="iic/SenseVoiceCTC" |
|---|
| | | model_name_or_model_dir="iic/SenseVoiceSmall" |
|---|
| | | |
|---|
| | | ## option 2, download model by git |
|---|
| | | #local_path_root=${workspace}/modelscope_models |
|---|
| | |
|---|
| | | echo $DISTRIBUTED_ARGS |
|---|
| | | |
|---|
| | | # funasr trainer path |
|---|
| | | train_tool=`dirname $(which funasr)`/train_ds.py |
|---|
| | | train_tool=../../../funasr/bin/train_ds.py |
|---|
| | | |
|---|
| | | torchrun $DISTRIBUTED_ARGS \ |
|---|
| | | ${train_tool} \ |
|---|
| | |
|---|
| | | ++train_conf.use_deepspeed=false \ |
|---|
| | | ++train_conf.deepspeed_config=${deepspeed_config} \ |
|---|
| | | ++optim_conf.lr=0.0002 \ |
|---|
| | | ++output_dir="${output_dir}" &> ${log_file} |
|---|
| | | ++output_dir="${output_dir}" &> ${log_file} |
|---|
