Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 9ac7187

Browse files
committedApr 13, 2023
Add finetune evaluation script
1 parent d741785 commit 9ac7187

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed
 

‎ptuning/evaluate_finetune.sh

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
CHECKPOINT=adgen-chatglm-6b-ft-1e-4
2+
STEP=3000
3+
4+
CUDA_VISIBLE_DEVICES=0 python3 main.py \
5+
--do_predict \
6+
--validation_file AdvertiseGen/dev.json \
7+
--test_file AdvertiseGen/dev.json \
8+
--overwrite_cache \
9+
--prompt_column content \
10+
--response_column summary \
11+
--model_name_or_path ./output/$CHECKPOINT/checkpoint-$STEP \
12+
--output_dir ./output/$CHECKPOINT \
13+
--overwrite_output_dir \
14+
--max_source_length 256 \
15+
--max_target_length 256 \
16+
--per_device_eval_batch_size 1 \
17+
--predict_with_generate \
18+
--fp16_full_eval

0 commit comments

Comments
 (0)
Please sign in to comment.