bc-models-deberta / train_config.json
slamos's picture
Upload train_config.json with huggingface_hub
5a9231c verified
{
"model_type": "deberta",
"pretrained": "microsoft/deberta-v3-base",
"epochs": 5,
"batch_size": 16,
"lr": 1e-05,
"max_length": 512,
"class_weights": [
1.0,
3.5319032669067383,
10.121383666992188
],
"num_labels": 3,
"id2label": {
"0": "SAME_PARAGRAPH",
"1": "NEW_PARAGRAPH",
"2": "NEWLINE"
},
"label2id": {
"SAME_PARAGRAPH": 0,
"NEW_PARAGRAPH": 1,
"NEWLINE": 2
}
}