Skip to content

Commit

Permalink
update orpo shell.
Browse files Browse the repository at this point in the history
  • Loading branch information
shibing624 committed Apr 17, 2024
1 parent 1a61aa9 commit a6e9647
Showing 1 changed file with 30 additions and 0 deletions.
30 changes: 30 additions & 0 deletions run_orpo.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
CUDA_VISIBLE_DEVICES=0,1 python orpo_training.py \
--model_type bloom \
--model_name_or_path bigscience/bloomz-560m \
--train_file_dir ./data/reward \
--validation_file_dir ./data/reward \
--per_device_train_batch_size 4 \
--per_device_eval_batch_size 1 \
--do_train \
--do_eval \
--use_peft True \
--max_train_samples 1000 \
--max_eval_samples 10 \
--max_steps 100 \
--eval_steps 20 \
--save_steps 50 \
--max_source_length 128 \
--max_target_length 128 \
--output_dir outputs-orpo-bloom-v1 \
--target_modules all \
--lora_rank 8 \
--lora_alpha 16 \
--lora_dropout 0.05 \
--torch_dtype float16 \
--fp16 True \
--device_map auto \
--report_to tensorboard \
--remove_unused_columns False \
--gradient_checkpointing True \
--orpo_beta 0.1 \
--cache_dir ./cache

0 comments on commit a6e9647

Please sign in to comment.