| { | |
| "architectures": [ | |
| "TotalClassifierModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration.TotalClassifierConfig", | |
| "AutoModel": "modeling.TotalClassifierModel" | |
| }, | |
| "backbone": "tf_efficientnetv2_b0", | |
| "cnn_dropout": 0.1, | |
| "feature_dim": 192, | |
| "image_size": [ | |
| 256, | |
| 256 | |
| ], | |
| "in_chans": 1, | |
| "linear_dropout": 0.1, | |
| "model_type": "total_classifier", | |
| "num_classes": 117, | |
| "rnn_dropout": 0.0, | |
| "rnn_num_layers": 1, | |
| "rnn_type": "GRU", | |
| "seq_len": 512, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.47.0" | |
| } | |