update

yuchenlin · Apr 1, 2024 · de20acb · de20acb
1 parent 3423852
commit de20acb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -1,3 +1,6 @@
 [submodule "data/UnifiedFeedback"]
 	path = data/UnifiedFeedback
-	url = https://github.com/jdf-prog/UnifiedFeedback.git
+	url = https://github.com/jdf-prog/UnifiedFeedback
+[submodule "hf_space/LLM-Blender"]
+	path = hf_space/LLM-Blender
+	url = https://huggingface.co/spaces/llm-blender/LLM-Blender
diff --git a/train_ranker.sh b/train_ranker.sh
@@ -15,7 +15,7 @@ dataset="UnifiedFeedback"
 eval_dataset="reward_bench"
 backbone_type="phi" # "deberta" or "roberta"
 backbone_name="microsoft/phi-2" # "microsoft/deberta-v3-large" or "roberta-large"
-n_gpu=4
+n_gpu=8
 ranker="PairRanker" # "PairRanker" or "Summareranker" or "SimCLS"
 candidate_model="" # separted by comma. Empty string for all models
 candidate_decoding_method="" # separted by comma. Empty string for all methods