{
  "model_name": "universal-upscale-distilled",
  "task": "image-super-resolution",
  "native_scale": 9,
  "parameters": 1296013,
  "parameters_millions": 1.296,
  "architecture": {
    "model_type": "student_sr",
    "in_channels": 3,
    "out_channels": 3,
    "num_features": 64,
    "num_blocks": 8,
    "scale_factor": 9,
    "upsample_mode": "staged",
    "upsample_stages": [
      3,
      3
    ]
  },
  "onnx": {
    "file": "model.onnx",
    "file_bytes": 5197024,
    "opset": {
      "ai.onnx": 17
    },
    "ir_version": 10,
    "producer": "pytorch 2.11.0+cu130",
    "nodes": 42,
    "ops": [
      "Add",
      "Conv",
      "DepthToSpace",
      "PRelu"
    ],
    "initializers": 52,
    "external_data_tensors": 0,
    "self_contained": true
  },
  "input": {
    "name": "input",
    "dtype": "float32",
    "layout": "NCHW",
    "channels": "RGB",
    "range": "[0, 1]",
    "dynamic_axes": [
      "batch",
      "height",
      "width"
    ]
  },
  "output": {
    "name": "output",
    "dtype": "float32",
    "layout": "NCHW",
    "channels": "RGB",
    "shape": "[batch, 3, 9 * height, 9 * width]"
  },
  "training": {
    "method": "knowledge distillation",
    "teacher": "realesrgan_x4plus",
    "kd_alpha": 0.6,
    "target_source": "both",
    "iters": 4000,
    "seconds": 2932.2,
    "best_val_psnr": 28.695699244559165,
    "loss": {
      "l1_weight": 1.0,
      "perceptual_weight": 0.1,
      "edge_weight": 0.0
    },
    "note": "Validation PSNR is measured on the notebook's own validation crops; examples/verification.json measures the published ONNX on local DIV2K crops."
  },
  "verification": {
    "verdict": "ok",
    "torchscript_vs_onnx_max_abs_diff": 1.3113021850585938e-06,
    "report": "examples/verification.json"
  },
  "related": {
    "non_distilled_repo": "tailot/universal-upscale",
    "hub_repo": "tailot/universal-upscale-distilled"
  },
  "generated_at": "2026-10-25T22:25:42"
}