{ | |
"architectures": [ | |
"TransformerClassifier" | |
], | |
"auto_map": { | |
"AutoConfig": "transformer.TransformerClassifierConfig", | |
"AutoModel": "transformer.TransformerClassifier" | |
}, | |
"d_model": 128, | |
"ff_dim": 512, | |
"in_dim": 42, | |
"model_type": "transformer-checker", | |
"n_classes": 2, | |
"n_heads": 8, | |
"n_layers": 6, | |
"torch_dtype": "float32", | |
"transformers_version": "4.40.1" | |
} | |