1earner1 commited on
Commit
dc20024
·
verified ·
1 Parent(s): 7f61966

Upload run_mix_lora_config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. run_mix_lora_config.py +33 -0
run_mix_lora_config.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ model_name="mistralai/Mistral-7B-Instruct-v0.3" #"meta-llama/CodeLlama-7b-hf" #
3
+ load_path="./HPC_25_9_mistral_CN_catiffp_20k_lora_900/checkpoint_0_4000"
4
+ traindataset_files=['./dataset_lfs/CN_python800_catifelse_train_pairs.json']#'./dataset_lfs/CC_900_py_catifelse_train2_pairs.json']
5
+ #[ "./dataset_lfs/CC_900_py_catiffp_train2_pairs.json",]
6
+ #"./dataset_lfs/allpairs_data_large2_900_unperturbed_filtered.json",
7
+ #"./dataset_lfs/allpairs_data_large_900_var_rename_one.json",
8
+ #"./dataset_lfs/allpairs_data_large_900_loop2.json", ]
9
+ # "./dataset_lfs/allpairs_data_large_900_iffp2.json", ]
10
+ # "./dataset_lfs/allpairs_data_large_900_ind.json" ] #var_rename_three
11
+ valdataset_files=[ "./dataset_lfs/allpairs_data_val_900_unperturbed.json",
12
+ #"./dataset_lfs/allpairs_data_val_900_var_rename_one.json",
13
+ #"./dataset_lfs/allpairs_data_val_900_loop2.json",]
14
+ "./dataset_lfs/allpairs_data_val_900_iffp2.json", ]
15
+ #"./dataset_lfs/allpairs_data_val_900_ind.json"]
16
+ testdataset_files=[ './llm_for_code/datasets/codecontests/CC_py_900_catifelse_pairs.json'
17
+ #"./llm_for_code/datasets/codecontests/verified_var_rename_one_900.json",
18
+ #"./llm_for_code/datasets/codecontests/verified_loop2_900.json",]
19
+ # "./llm_for_code/datasets/codecontests/verified_iffp2_900.json",]
20
+ # "./llm_for_code/datasets/codecontests/verified_ind_900.json"
21
+ ]
22
+ category_included=['0_1_1' , '0_1_0', '0_2_0', '1_1_1', '1_2_1', '1_0_1', '1_1_0', '1_2_0', '2_2_0', '2_1_0', '2_0_0']
23
+ #category_included = ['0_1_1', '0_1_0']
24
+ initial_lr=5e-7
25
+ checkpoint_store_dir_path="./HPC_25_9_mistral_CN_catiffp2_20k_lora_900"
26
+ num_epochs=5
27
+ batch_size_train=1
28
+ max_length=2000
29
+ ckpnt_NUM=2000
30
+ SAVEALL=0
31
+ max_new_tok=50
32
+ prompt_file_path='prompt_only_instruct.txt' #'prompt_var_rename.txt' #'./loop_prompt.txt' CUDA_VISIBLE_DEVICES=1
33
+ #CUDA_VISIBLE_DEVICES=1 python run_lora_mistral_loop.py --model_name $model_name --traindataset_file $traindataset_file --valdataset_file $valdataset_file --testdataset_file $testdataset_file --checkpoint_store_dir_path $checkpoint_store_dir_path --initial_lr $initial_lr --num_epochs $num_epochs --batch_size_train $batch_size_train --ckpnt_num $ckpnt_num --saveall $saveall --prompt_file_path $prompt_file_path --max_length $max_length --max_new_tok $max_new_tok