config.json

{
  "architectures": [
    "Med3DSeg"
  ],
  "dice_weight": 1.0,
  "embed_dim": 768,
  "focal_weight": 1.0,
  "image_size": [
    128,
    256,
    256
  ],
  "mlp_dim": 2048,
  "model_type": "med3d_seg",
  "num_heads": 8,
  "pass_num": 2,
  "patch_size": [
    64,
    64,
    64
  ],
  "text_model": "nlpie/tiny-clinicalbert",
  "torch_dtype": "bfloat16",
  "transformer_depth": 2,
  "transformers_version": "4.51.0",
  "vision_encoder": "",
  "wb_project": "Med3DSeg"
}