-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_train.sh
10 lines (8 loc) · 872 Bytes
/
run_train.sh
1
2
3
4
5
6
7
8
9
10
/home/jiaanwang/anaconda3/envs/glm/bin/python3.8 train.py \
--model-type xglm \
--load_model_dir /home/fengwang/ASR/model/xglm1.7b \
--save_train_model_file /home/fengwang/ASR/XGLM/checkpoint
# /home/jiaanwang/anaconda3/envs/glm/bin/python3.8 -m torch.distributed.launch --nproc_per_node 2 train.py \
# --model-type glm \
# --load_model_dir /home/fengwang/ASR/model/bloom_3b \
# --save_train_model_file /home/fengwang/ASR/XGLM/checkpoint