diff --git a/lib/style-transfer/bash/experiment/rb_gen/az/llama3.2-3b-pb-seed-replaced-small.sh b/lib/style-transfer/bash/experiment/rb_gen/az/llama3.2-3b-pb-seed-replaced-small.sh new file mode 100644 index 0000000..e452a6e --- /dev/null +++ b/lib/style-transfer/bash/experiment/rb_gen/az/llama3.2-3b-pb-seed-replaced-small.sh @@ -0,0 +1,15 @@ +export CUDA_VISIBLE_DEVICES=$1 +python style_transfer/run_rb_gen.py model.name=meta-llama/Llama-3.2-3B-Instruct \ + model.peft_config.target_modules='["q_proj", "k_proj", "v_proj", "o_proj", "gate_proj", "up_proj", "down_proj"]' \ + dataset.name=bio-datasets/mimic-iii-gpt4o-tokens \ + max_steps=7 \ + dataset.num_generated_samples=1000 \ + score.model.model_name_or_path=sentence-transformers/all-mpnet-base-v2 \ + dataset.sft_ratio=0.06 \ + dataset.gen_ratio=0.7 \ + sft.training_args.eval_steps=30 \ + score.train.train_size=0.6 \ + dataset.sft_dataset.size=977 \ + dpo.training_args.num_train_epochs=20 \ + dpo.percentile=70 \ + score.batch_size=8