@echo off set PRE_SEQ_LEN=128 set CHECKPOINT=adgen-chatglm2-6b-pt-128-2e-2 set STEP=3000 set NUM_GPUS=1

torchrun --standalone --nnodes=1 --nproc-per-node=%NUM_GPUS% main.py ^ --do_predict ^ --validation_file AdvertiseGen/dev.json ^ --test_file AdvertiseGen/dev.json ^ --overwrite_cache ^ --prompt_column content ^ --response_column summary ^ --model_name_or_path THUDM/chatglm2-6b ^ --ptuning_checkpoint ./output/%CHECKPOINT%/checkpoint-%STEP% ^ --output_dir ./output/%CHECKPOINT% ^ --overwrite_output_dir ^ --max_source_length 64 ^ --max_target_length 64 ^ --per_device_eval_batch_size 1 ^ --predict_with_generate ^ --pre_seq_len %PRE_SEQ_LEN% ^ --quantization_bit

PRE_SEQ_LEN=128CHECKPOINT=adgen-chatglm2-6b-pt-128-2e-2STEP=3000NUM_GPUS=1torchrun --standalone --nnodes=1 --nproc-per-node=$NUM_GPUS mainpy --do_predict --validation_file AdvertiseGendevjson

原文地址: https://www.cveoy.top/t/topic/iCRH 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录