2021-12-23 10:21:27 +01:00
|
|
|
#MODEL = '../MODELS/without_date/checkpoint-395000'
|
|
|
|
MODEL = 'roberta-base'
|
2021-12-28 12:05:47 +01:00
|
|
|
BATCH_SIZE = 90
|
2021-12-23 10:21:27 +01:00
|
|
|
EARLY_STOPPING = 3
|
2021-12-28 12:05:47 +01:00
|
|
|
WARMUP_STEPS = 5_000
|
2021-12-23 10:21:27 +01:00
|
|
|
LR=1e-6
|
|
|
|
NUM_EPOCHS = 20
|
|
|
|
STEPS_EVAL = 5_000
|
|
|
|
TEST=False
|