Resnet50_backend / config.json
Danieli1021's picture
Update config.json
07fa5a7 verified
{
"architectures": [
"RoadworkBinaryClassifier"
],
"model_type": "custom-resnet50-binary-classifier",
"backbone_name": "resnet50",
"num_labels": 1,
"id2label": {
"0": "None",
"1": "Roadwork"
},
"label2id": {
"None": 0,
"Roadwork": 1
},
"problem_type": "binary_classification",
"classifier_activation": "sigmoid",
"decision_threshold": 0.5,
"image_size": 224,
"torch_dtype": "float32",
"best_val_f1": 0.9349030470914127,
"best_val_validator_reward_proxy": 0.5447887084945165
}