models/unet_plus_plus/config.json

{
  "architectures": [
    "HFUnetPlusPlus"
  ],
  "auto_map": {
    "AutoConfig": "hf_config.UnetConfig",
    "AutoModelForImageSegmentation": "hf_model.HFUnetPlusPlus"
  },
  "decoder_channels": [
    1024,
    512,
    256,
    128,
    64
  ],
  "encoder_name": "resnet18",
  "input_channels": 1,
  "num_classes": 16,
  "torch_dtype": "float32",
  "transformers_version": "4.24.0"
}