config.json

{
  "data": {
    "image_shape": [
      462,
      616
    ]
  },
  "generic": {
    "seed": 13
  },
  "model": {
    "expansion": 4,
    "name": "UniDepthV1",
    "num_heads": 8,
    "pixel_decoder": {
      "depths": [
        3,
        2,
        1
      ],
      "dropout": 0.0,
      "hidden_dim": 512
    },
    "pixel_encoder": {
      "depths": [
        3,
        6,
        33,
        36
      ],
      "embed_dim": 192,
      "embed_dims": [
        192,
        192,
        192,
        384,
        384,
        384,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        768,
        1536,
        1536,
        1536
      ],
      "name": "convnext_large",
      "patch_size": 16,
      "pretrained": null
    }
  },
  "training": {}
}