Test_LSTM_BERT / config.json
UNCANNY69's picture
Update config.json
114f83c verified
raw
history blame contribute delete
546 Bytes
{
"architectures": [
"BertLSTMForSequenceClassification"
],
"auto_map": {
"AutoConfig": "model.BertLSTMConfig",
"AutoModelForSequenceClassification": "model.BertLSTMForSequenceClassification"
},
"dropout_rate": 0.1,
"embed_dim": 768,
"hidden_dim_lstm": 256,
"id2label": {
"0": "fake",
"1": "true"
},
"label2id": {
"fake": 0,
"true": 1
},
"model_type": "bertLSTMForSequenceClassification",
"num_classes": 2,
"num_layers": 12,
"torch_dtype": "float32",
"transformers_version": "4.38.2"
}