model_config.json

{
  "checkpoint_source": "epoch=17-step=1836.ckpt",
  "created_at": "2025-11-23T17:24:34.909430Z",
  "hparams": {
    "backbone_type": "dinov3",
    "in_ch": 1,
    "loss": "<class 'odelia.models.utils.losses.MulitCELoss'>",
    "loss_kwargs": {
      "class_labels_num": [
        3
      ]
    },
    "lr_scheduler": null,
    "lr_scheduler_kwargs": {},
    "model_size": "s",
    "optimizer": "<class 'torch.optim.adamw.AdamW'>",
    "optimizer_kwargs": {
      "lr": 1e-05
    },
    "out_ch": 3,
    "save_hyperparameters": true,
    "slice_fusion_type": "transformer",
    "spatial_dims": 3
  },
  "model_class": "odelia.models.mst.MSTRegression"
}