Skip to content

Commit

Permalink
Update train.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
shuo-git authored Oct 9, 2024
1 parent 0714ed9 commit 74c3826
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions scripts/train.sh
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,4 @@ OPTS+=" --lora_root_path ./checkpoints/BM_LoRAs"
OPTS+=" --train_data_path ./data/train/${language}_${task}_train.jsonl"
OPTS+=" --batch_size_per_device $bsz --epochs $epoch --lr $lr"


# 运行任务
torchrun --nnodes=1 --nproc_per_node=2 --master_port=$MASTER_PORT $CODE/train.py ${OPTS} &> $LOG_FILE

0 comments on commit 74c3826

Please sign in to comment.