Skip to content

Commit

Permalink
Update run.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
XiaoxinHe authored Mar 6, 2024
1 parent 4cee840 commit 9812ee2
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@ do
# 1) inference only
# a) Question-Only
python inference.py --dataset expla_graphs --model_name inference_llm --llm_model_name 7b_chat --max_txt_len 0 --seed $seed
python inference.py --dataset scene_graphs --model_name inference_llm --llm_model_name 7b_chat --max_txt_len 0 --seed $seed
python inference.py --dataset webqsp --model_name inference_llm --llm_model_name 7b_chat --max_txt_len 0 --seed $seed
# b) Textual Graph + Question
python inference.py --dataset expla_graphs --model_name inference_llm --llm_model_name 7b_chat --seed $seed
Expand All @@ -25,9 +24,8 @@ python train.py --dataset webqsp --model_name graph_llm --seed $seed
python train.py --dataset expla_graphs --model_name llm --llm_frozen False --seed $seed
python train.py --dataset scene_graphs_baseline --model_name llm --llm_frozen False --seed $seed
python train.py --dataset webqsp_baseline --model_name llm --llm_frozen False --seed $seed

# b) g-retriever + finetuning with lora
python train.py --dataset expla_graphs --model_name graph_llm --llm_frozen False --seed $seed
python train.py --dataset scene_graphs --model_name graph_llm --llm_frozen False --seed $seed
python train.py --dataset webqsp --model_name graph_llm --llm_frozen False --seed $seed
done
done

0 comments on commit 9812ee2

Please sign in to comment.