ireland-news-headlines-year.../hf_roberta_base_classification/config.py

12 lines
212 B
Python

#MODEL = '../MODELS/without_date/checkpoint-395000'
MODEL = 'roberta-base'
BATCH_SIZE = 90
EARLY_STOPPING = 3
WARMUP_STEPS = 5_000
LR=1e-5
NUM_EPOCHS = 20
STEPS_EVAL = 500
TEST=False
MIN_YEAR=1996
MAX_YEAR=2019