code_for_ade20k/checkpoint-32800/unet_ema/config.json

{
  "SPADE_type": "spade",
  "_class_name": "UNetModel",
  "_diffusers_version": "0.16.1",
  "attention_resolutions": [
    2,
    4,
    8
  ],
  "channel_mult": [
    1,
    2,
    3,
    4
  ],
  "conv_resample": true,
  "decay": 0.9999,
  "dims": 2,
  "dropout": 0,
  "image_size": [
    64,
    64
  ],
  "in_channels": 4,
  "inv_gamma": 1.0,
  "mask_emb": "resize",
  "min_decay": 0.0,
  "model_channels": 256,
  "num_classes": 151,
  "num_head_channels": -1,
  "num_heads": 8,
  "num_heads_upsample": -1,
  "num_res_blocks": 2,
  "optimization_step": 32800,
  "out_channels": 4,
  "power": 0.75,
  "resblock_updown": true,
  "update_after_step": 0,
  "use_checkpoint": true,
  "use_ema_warmup": true,
  "use_fp16": true,
  "use_new_attention_order": false,
  "use_scale_shift_norm": true
}