chatglm-6b-int4 / ptuning /train_linux.sh
NewBreaker's picture
first
23aa310
PRE_SEQ_LEN=128
LR=2e-2
CUDA_VISIBLE_DEVICES=0 python main.py \
--do_train \
--train_file ../AdvertiseGen/train.json \
--validation_file ../AdvertiseGen/dev.json \
--prompt_column content \
--response_column summary \
--overwrite_cache \
--model_name_or_path ../models/chatglm-6b-int4/models--THUDM--chatglm-6b-int4/snapshots/e02ba894cf18f3fd9b2526c795f983683c4ec732 \
--output_dir output/adgen-chatglm-6b-pt-$PRE_SEQ_LEN-$LR \
--overwrite_output_dir \
--max_source_length 64 \
--max_target_length 64 \
--per_device_train_batch_size 1 \
--per_device_eval_batch_size 1 \
--gradient_accumulation_steps 16 \
--predict_with_generate \
--max_steps 3000 \
--logging_steps 10 \
--save_steps 1000 \
--learning_rate $LR \
--pre_seq_len 1 \
--quantization_bit 4