SwinFusion/config.json

{
  "Ex_depths": [
    4
  ],
  "Ex_num_heads": [
    6,
    6
  ],
  "Fusion_depths": [
    2,
    2
  ],
  "Fusion_num_heads": [
    6,
    6,
    6
  ],
  "Re_depths": [
    4
  ],
  "Re_num_heads": [
    6,
    6
  ],
  "ape": false,
  "architectures": [
    "SwinFusion"
  ],
  "attn_drop_rate": 0.0,
  "drop_path_rate": 0.1,
  "drop_rate": 0.0,
  "embed_dim": 60,
  "img_range": 1.0,
  "img_size": 128,
  "in_chans": 1,
  "mlp_ratio": 2.0,
  "model_type": "swin-fusion",
  "patch_norm": true,
  "patch_size": 1,
  "qk_scale": null,
  "qkv_bias": true,
  "resi_connection": "1conv",
  "torch_dtype": "float32",
  "transformers_version": "4.47.0",
  "upsampler": null,
  "upscale": 1,
  "use_checkpoint": false,
  "window_size": 8
}