roberta regressor head fairseq 12 eopch bs=25

2021-09-19 11:43:31 +02:00 · 2021-09-19 11:43:31 +02:00 · 632b5dff50
parent dd9d020838
commit 632b5dff50
3 changed files with 297379 additions and 297379 deletions
--- a/dev-0/out.tsv
+++ b/dev-0/out.tsv
--- a/roberta_regressor_head/train.py
+++ b/roberta_regressor_head/train.py
@ -12,7 +12,7 @@ from sklearn.preprocessing import MinMaxScaler

 EVAL_OFTEN = True
 EVAL_EVERY = 3000
-BATCH_SIZE = 1
+BATCH_SIZE = 25
 model_type = 'base' # base or large


--- a/test-A/out.tsv
+++ b/test-A/out.tsv