kubapok/ireland-news-headlines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

t5 v1.1. 5.93 epochs

Browse Source
This commit is contained in:
kubapok 2021-10-04 17:28:22 +02:00
parent 2afdec1393
commit 934cc0e5be
4 changed files with 56501 additions and 56500 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86270
dev-0/out.tsv
View File

File diff suppressed because it is too large Load Diff

2
t5_no_year/04_predict.py
View File

@ -4,7 +4,7 @@ from tqdm import tqdm
from config import LABELS_LIST from config import LABELS_LIST
device = 'cuda' device = 'cuda'
model_path= 't5-retrained' model_path= 't5-retrained/checkpoint-110000'
from transformers import AutoModelForSequenceClassification from transformers import AutoModelForSequenceClassification

1
t5_no_year/run.sh
View File

@ -7,6 +7,7 @@ python run_summarization.py \
--summary_column 'label' \ --summary_column 'label' \
--max_target_length=4 \ --max_target_length=4 \
--max_source_length=64 \ --max_source_length=64 \
--num_train_epochs 20 \
--output_dir ./t5-retrained \ --output_dir ./t5-retrained \
--per_device_train_batch_size=16 \ --per_device_train_batch_size=16 \
--per_device_eval_batch_size=16 \ --per_device_eval_batch_size=16 \

26728
test-A/out.tsv
View File

File diff suppressed because it is too large Load Diff