Skip to content

GitLab

Commit e07b8855 authored by zhaoying1's avatar zhaoying1
Browse files

Update evaluate_ft.sh

parent c6186d00
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptuning/evaluate_ft.sh
View file @ e07b8855
...@@ -2,7 +2,7 @@ PRE_SEQ_LEN=128 ...@@ -2,7 +2,7 @@ PRE_SEQ_LEN=128
CHECKPOINT=adgen-chatglm-6b-ft-4c-5e-5 CHECKPOINT=adgen-chatglm-6b-ft-4c-5e-5
STEP=5000 STEP=5000
CUDA_VISIBLE_DEVICES=0 python3 main.py \ HIP_VISIBLE_DEVICES=0 python3 main.py \
--do_predict \ --do_predict \
--validation_file AdvertiseGen/dev.json \ --validation_file AdvertiseGen/dev.json \
--test_file AdvertiseGen/dev.json \ --test_file AdvertiseGen/dev.json \
...@@ -16,4 +16,4 @@ CUDA_VISIBLE_DEVICES=0 python3 main.py \ ...@@ -16,4 +16,4 @@ CUDA_VISIBLE_DEVICES=0 python3 main.py \
--max_target_length 64 \ --max_target_length 64 \
--per_device_eval_batch_size 1 \ --per_device_eval_batch_size 1 \
--predict_with_generate \ --predict_with_generate \
--pre_seq_len $PRE_SEQ_LEN --pre_seq_len $PRE_SEQ_LEN
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment