Add some results

2023-05-16 16:53:03 +08:00
parent 737cfa1c31
commit 479a9e26fe
7 changed files with 25 additions and 6 deletions
--- a/src/scripts/train.sh
+++ b/src/scripts/train.sh
@ -1,6 +1,6 @@
 #!/bin/bash

-WORLD_SIZE=8 CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nproc_per_node=8 --master_port=1235 train.py \
+WORLD_SIZE=8 CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nproc_per_node=8 --master_port=1235 train_lora.py \
    --base_model '../models/base_models/chinese_llama_7b' \
    --data_path '' \
    --output_dir '../models/lora_weights' \
--- a/src/train_lora.py
+++ b/src/train_lora.py