We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d741785 commit 9ac7187Copy full SHA for 9ac7187
ptuning/evaluate_finetune.sh
@@ -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