config.json

{
  "generic": {
    "seed": 13
  },
  "training": {},
  "data": {
    "image_shape": [
      462,
      616
    ]
  },
  "model": {
    "name": "UniDepthV1",
    "num_heads": 8,
    "expansion": 4,
    "pixel_decoder": {
      "hidden_dim": 512,
      "depths": [
        3,
        2,
        1
      ],
      "dropout": 0.0
    },
    "pixel_encoder": {
      "name": "convnext_large",
      "pretrained": null,
      "patch_size": 16,
      "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
      ],
      "depths": [
        3,
        6,
        33,
        36
      ]
    }
  }
}