| { | |
| "architecture": "resnet18", | |
| "architectures": [ | |
| "HFResNet18Detector" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "hf_config.HFResNet18DetectorConfig", | |
| "AutoModelForImageClassification": "hf_model.HFResNet18Detector" | |
| }, | |
| "dropout_rate": 0.5, | |
| "freeze_backbone": false, | |
| "image_size": 224, | |
| "label_smoothing": 0.1, | |
| "max_grad_norm": 1.0, | |
| "model_type": "resnet18-detector", | |
| "num_classes": 2, | |
| "pretrained_weights": "IMAGENET1K_V1", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.54.1", | |
| "weight_decay": 0.1 | |
| } | |