Update train_model.sh
Browse files- train_model.sh +1 -1
train_model.sh
CHANGED
|
@@ -4,7 +4,7 @@ torchrun --nproc_per_node=1 --master_port=9292 train.py \
|
|
| 4 |
--tokenizer_name_or_path /src/weights/tokenizer \
|
| 5 |
--data_path ./alpaca_data.json \
|
| 6 |
--model_name_or_path /src/weights/llama-7b \
|
| 7 |
-
|
| 8 |
--output_dir alpaca_out \
|
| 9 |
--num_train_epochs 3 \
|
| 10 |
--per_device_train_batch_size 4 \
|
|
|
|
| 4 |
--tokenizer_name_or_path /src/weights/tokenizer \
|
| 5 |
--data_path ./alpaca_data.json \
|
| 6 |
--model_name_or_path /src/weights/llama-7b \
|
| 7 |
+
#--bf16 True \
|
| 8 |
--output_dir alpaca_out \
|
| 9 |
--num_train_epochs 3 \
|
| 10 |
--per_device_train_batch_size 4 \
|