t5 v1.1. 5.93 epochs

2021-10-04 17:28:22 +02:00 · 2021-10-04 17:28:22 +02:00 · 934cc0e5be
commit 934cc0e5be
parent 2afdec1393
4 changed files with 56501 additions and 56500 deletions
--- a/dev-0/out.tsv
+++ b/dev-0/out.tsv
--- a/t5_no_year/04_predict.py
+++ b/t5_no_year/04_predict.py
@ -4,7 +4,7 @@ from tqdm import tqdm
 from config import LABELS_LIST

 device = 'cuda'
-model_path= 't5-retrained'
+model_path= 't5-retrained/checkpoint-110000'


 from transformers import AutoModelForSequenceClassification
--- a/t5_no_year/run.sh
+++ b/t5_no_year/run.sh
@ -7,6 +7,7 @@ python run_summarization.py \
    --summary_column 'label' \
    --max_target_length=4 \
    --max_source_length=64 \
+    --num_train_epochs 20 \
    --output_dir ./t5-retrained \
    --per_device_train_batch_size=16 \
    --per_device_eval_batch_size=16 \
--- a/test-A/out.tsv
+++ b/test-A/out.tsv