config.py

params = {
    'pretrained_model_name': 'ltgoslo/norbert2',
    'path_to_model_bin': 'saved_models/norbert2_epoch_5.bin',
    'LR': 1e-05,
    'dropout': 0.4,
    'warmup': 2,
    'epochs': 10,
    'max_length': 512,
    'batch_size': 4,
}