"src/vscode:/vscode.git/clone" did not exist on "5755d16868ec3da7d5eb4f42db77b01fac842ea8"
Commit e07b8855 authored by zhaoying1's avatar zhaoying1
Browse files

Update evaluate_ft.sh

parent c6186d00
......@@ -2,7 +2,7 @@ PRE_SEQ_LEN=128
CHECKPOINT=adgen-chatglm-6b-ft-4c-5e-5
STEP=5000
CUDA_VISIBLE_DEVICES=0 python3 main.py \
HIP_VISIBLE_DEVICES=0 python3 main.py \
--do_predict \
--validation_file AdvertiseGen/dev.json \
--test_file AdvertiseGen/dev.json \
......@@ -16,4 +16,4 @@ CUDA_VISIBLE_DEVICES=0 python3 main.py \
--max_target_length 64 \
--per_device_eval_batch_size 1 \
--predict_with_generate \
--pre_seq_len $PRE_SEQ_LEN
\ No newline at end of file
--pre_seq_len $PRE_SEQ_LEN
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