bakaEC/kopa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #28 from zjukg/master

Browse Source 
change files and links
This commit is contained in:
Yichi Zhang 2023-12-27 17:16:53 +08:00 committed by GitHub
commit df16aad012
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/run_umls_kopa.sh → scripts/run_codex_kopa.sh
View File

@ -2,13 +2,13 @@ export WANDB_DISABLED=true
wandb offline
CUDA_VISIBLE_DEVICES=0 nohup python finetune_kopa.py \
--base_model 'YOUR LLM PATH' \
--data_path 'data/UMLS-train.json' \
--data_path 'data/CoDeX-S-train.json' \
--output_dir 'YOUR SAVE PATH' \
--num_epochs 3 \
--lora_r 32 \
--lora_r 64 \
--learning_rate 3e-4 \
--batch_size 12 \
--micro_batch_size 12 \
--num_prefix 1 \
--kge_model 'data/UMLS-rotate.pth' \
--kge_model 'data/CoDeX-S-rotate.pth' \
--lora_target_modules='[q_proj,k_proj,v_proj,o_proj]' > log.txt &

11
scripts/run_umls_base.sh → scripts/run_fb15k237n_kopa.sh
View File

@ -1,13 +1,14 @@
export WANDB_DISABLED=true
wandb offline
CUDA_VISIBLE_DEVICES=1 nohup python finetune.py \
CUDA_VISIBLE_DEVICES=0 nohup python finetune_kopa.py \
--base_model 'YOUR LLM PATH' \
--data_path 'preprocess/UMLS-train.json' \
--data_path 'data/FB15K-237N-train.json' \
--output_dir 'YOUR SAVE PATH' \
--num_epochs 3 \
--lora_r 64 \
--learning_rate 3e-4 \
--batch_size 12 \
--micro_batch_size 12 \
--lora_r 32 \
--learning_rate 3e-4 \
--num_prefix 1 \
--kge_model 'data/FB15K-237N-rotate.pth' \
--lora_target_modules='[q_proj,k_proj,v_proj,o_proj]' > log.txt &