config.json

{
  "_name_or_path": "./cell_sribd/pytorch_model.bin",
  "architectures": [
    "MyModel"
  ],
  "block_size": 2048,
  "context": 128,
  "device": "cpu",
  "input_channels": 3,
  "ksize": 15,
  "min_overlap": 128,
  "model_type": "cell_sribd",
  "n_rays": 32,
  "np_thres": 0.6,
  "num_classes": 4,
  "obj_size_thres": 100,
  "overall_thres": 0.4,
  "overlap": 0.5,
  "roi_size": [
    [
      512,
      512
    ],
    [
      512,
      512
    ]
  ],
  "sw_batch_size": 4,
  "torch_dtype": "float32",
  "transformers_version": "4.27.1"
}