| { | |
| "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" | |
| } |