config.json

{
  "architectures": [
    "UVDocModel"
  ],
  "backbone_config": {
    "depths": [
      1,
      1
    ],
    "kernel_size": 5,
    "model_type": "uvdoc_backbone",
    "out_features": [
      "stage1",
      "stage2"
    ],
    "out_indices": [
      1,
      2
    ],
    "resnet_configs": [
      [
        [
          8,
          8,
          1,
          false
        ],
        [
          8,
          8,
          3,
          false
        ],
        [
          8,
          8,
          3,
          false
        ]
      ],
      [
        [
          8,
          16,
          1,
          true
        ],
        [
          16,
          16,
          3,
          false
        ],
        [
          16,
          16,
          3,
          false
        ],
        [
          16,
          16,
          3,
          false
        ]
      ],
      [
        [
          16,
          32,
          1,
          true
        ],
        [
          32,
          32,
          3,
          false
        ],
        [
          32,
          32,
          3,
          false
        ],
        [
          32,
          32,
          3,
          false
        ],
        [
          32,
          32,
          3,
          false
        ],
        [
          32,
          32,
          3,
          false
        ]
      ]
    ],
    "resnet_head": [
      [
        3,
        8
      ],
      [
        8,
        8
      ]
    ],
    "stage_configs": [
      [
        [
          32,
          1
        ]
      ],
      [
        [
          32,
          2
        ]
      ]
    ],
    "stage_names": [
      "stem",
      "stage1",
      "stage2"
    ]
  },
  "bridge_connector": [
    32,
    32
  ],
  "dtype": "float32",
  "hidden_act": "prelu",
  "kernel_size": 5,
  "model_type": "uvdoc",
  "out_point_positions2D": [
    [
      32,
      8
    ],
    [
      8,
      2
    ]
  ],
  "padding_mode": "reflect",
  "transformers_version": "5.16.0.dev0"
}