model/config.json

{
    "args": [],
    "kwargs": {
        "in_channels": 3,
        "num_classes": 20,
        "num_bounding_boxes": 2,
        "grid_size": 7,
        "mlp_size": 1024,
        "internal_size": 1024
    },
    "class_name": "ResNetYoloV1",
    "module_name": "vision_models_playground.models.segmentation.yolo_v1"
}