{ | |
"architectures": [ | |
"BiLSTM" | |
], | |
"auto_map": { | |
"AutoConfig": "blstm_config.BiLSTMConfig", | |
"AutoModel": "blstm_model.BiLSTM" | |
}, | |
"device": "cuda:0", | |
"dropout": 0.33, | |
"embed_dim": 100, | |
"hidden_dim": 256, | |
"num_layers": 1, | |
"output_dim": 128, | |
"predict_output": 10, | |
"torch_dtype": "float32", | |
"transformers_version": "4.27.2", | |
"vocab_size": 23626 | |
} | |