fix

txsun1997 · Oct 18, 2022 · 34fedb3 · 34fedb3
1 parent 4bd5b55
commit 34fedb3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -131,8 +131,18 @@ The following example add and train a debias adapter in the BERT-large of BERTSc
 cd Metric-Fairness/mitigating_bias/train/BERTScore
 mkdir ./logs
 pip install -r requirements.txt
-INPUT_PATH= train.tsv # your dataset path
-bash train_bert_score_bert_large.sh
+INPUT_PATH= train.tsv # your training set path
+python train_BERTScore.py
+    --model_type bert-large-uncased \
+    --adapter_name debiased-bertscore \
+    --lr 5e-4 \
+    --warmup 0.0 \
+    --batch_size 16 \
+    --n_epochs 4 \
+    --seed 42 \
+    --device cuda \
+    --logging_steps 100 \
+    --data_path ${INPUT_PATH}
 ```
 
 

diff --git a/mitigating_bias/train/BERTScore/train_bert_score_bert_large.sh b/mitigating_bias/train/BERTScore/train_bert_score_bert_large.sh