{
  "architecture": "Restormer",
  "in_channels": 3,
  "out_channels": 3,
  "dim": 48,
  "num_blocks": [2, 3, 3, 4],
  "num_refinement_blocks": 4,
  "heads": [1, 2, 4, 8],
  "ffn_expansion_factor": 2.66
}