| { | |
| "architectures": [ | |
| "MammoEnsemble" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration.MammoConfig", | |
| "AutoModel": "modeling.MammoEnsemble" | |
| }, | |
| "backbone": "tf_efficientnetv2_s", | |
| "dropout": 0.1, | |
| "feature_dim": 1280, | |
| "image_sizes": [ | |
| [ | |
| 2048, | |
| 1024 | |
| ], | |
| [ | |
| 1920, | |
| 1280 | |
| ], | |
| [ | |
| 1536, | |
| 1536 | |
| ] | |
| ], | |
| "in_chans": 1, | |
| "model_type": "mammo", | |
| "num_classes": 5, | |
| "num_models": 3, | |
| "pad_to_aspect_ratio": [ | |
| true, | |
| true, | |
| false | |
| ], | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.47.0" | |
| } | |