config.json

{
  "anchors_ratios": [
    [
      2,
      4
    ],
    [
      4,
      8
    ],
    [
      8,
      16
    ]
  ],
  "architectures": [
    "YoloModel"
  ],
  "feature_extrator_kwargs": {
    "features_only": true,
    "model_name": "resnet50",
    "out_indices": [
      2,
      3,
      4
    ],
    "pretrained": true
  },
  "fpn_out_channels": 256,
  "image_size": [
    840,
    840
  ],
  "lb_loss_weights": [
    10,
    1
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.33.3"
}