| { | |
| "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" | |
| } | |