| { | |
| "model_type": "liif", | |
| "architecture": { | |
| "type": "LIIFNet (simplified Local Implicit Image Function)", | |
| "num_feat": 64, | |
| "num_res": 16, | |
| "decoder_hidden": 64, | |
| "local_size": 3 | |
| }, | |
| "scales": { | |
| "arbitrary": true, | |
| "trained_range": [ | |
| 1.5, | |
| 6.0 | |
| ], | |
| "tested_max": 9.2 | |
| }, | |
| "parameters": 4603897, | |
| "input": { | |
| "format": "RGB float32 tensor, shape (B, 3, H, W), values in [0, 1]" | |
| }, | |
| "output": { | |
| "format": "RGB float32 tensor, shape (B, 3, target_h, target_w), values in [0, 1]" | |
| }, | |
| "inference": { | |
| "signature": "model(lr, target_h, target_w)", | |
| "tile_size": 256, | |
| "overlap": 16, | |
| "tiling": "overlapping tiles + feather blending, compiled inside model_scripted.pt", | |
| "max_scale": 9.2, | |
| "max_scale_note": "largest scale factor verified end-to-end for these weights; beyond it the model still runs but is extrapolating outside its training range" | |
| }, | |
| "training": { | |
| "stage": "2", | |
| "iterations": 4000, | |
| "batch_size": 2, | |
| "patch_size": 192, | |
| "seed": 42, | |
| "optimizer": "adam", | |
| "lr": 4e-05, | |
| "min_lr": 2e-05, | |
| "scheduler": "cosine", | |
| "grad_clip": 1.0, | |
| "amp": false, | |
| "losses": [ | |
| "L1", | |
| "edge (Sobel gradient)", | |
| "PatchGAN adversarial" | |
| ], | |
| "loss_weights": { | |
| "l1": 1.0, | |
| "edge": 0.1, | |
| "gan": 0.05 | |
| }, | |
| "num_buckets": 16, | |
| "init_from": "checkpoints/liif_v4/best_liif.pth", | |
| "dataset": "DIV2K train + Flickr2K (3450 HR images)", | |
| "degradation": "Real-ESRGAN-style (blur, noise, downscale, JPEG)", | |
| "checkpoint_step": 3500 | |
| }, | |
| "artifacts": { | |
| "torchscript": "model_scripted.pt", | |
| "weights": "model.safetensors", | |
| "note": "model_scripted.pt is self-contained: no source code is needed to run the model." | |
| }, | |
| "export": { | |
| "torch_version": "2.14.1+cu130", | |
| "exported_at": "2026-10-23T20:41:45+00:00", | |
| "method": "torch.jit.script + torch.jit.freeze", | |
| "verification": { | |
| "checks": [ | |
| { | |
| "input": [ | |
| 160, | |
| 106 | |
| ], | |
| "target": [ | |
| 320, | |
| 212 | |
| ], | |
| "output_shape": [ | |
| 1, | |
| 3, | |
| 212, | |
| 320 | |
| ], | |
| "max_abs_diff": 1.1920928955078125e-07 | |
| }, | |
| { | |
| "input": [ | |
| 160, | |
| 106 | |
| ], | |
| "target": [ | |
| 509, | |
| 335 | |
| ], | |
| "output_shape": [ | |
| 1, | |
| 3, | |
| 335, | |
| 509 | |
| ], | |
| "max_abs_diff": 1.1920928955078125e-07 | |
| } | |
| ], | |
| "max_abs_diff": 1.1920928955078125e-07, | |
| "forward_single_shape": [ | |
| 1, | |
| 3, | |
| 212, | |
| 320 | |
| ], | |
| "forward_single_ok": true, | |
| "passed": true, | |
| "cuda_shape_check": [ | |
| 1, | |
| 3, | |
| 212, | |
| 320 | |
| ], | |
| "cuda_finite": true | |
| }, | |
| "examples": { | |
| "source_image": "0801.png", | |
| "files": { | |
| "input.png": [ | |
| 320, | |
| 240 | |
| ], | |
| "output_4x.png": [ | |
| 1280, | |
| 960 | |
| ], | |
| "output_9.2x.jpg": [ | |
| 1472, | |
| 1104 | |
| ], | |
| "compare.jpg": [ | |
| 496, | |
| 142 | |
| ] | |
| } | |
| }, | |
| "demo": { | |
| "source_image": "a.png", | |
| "scale": 4.0, | |
| "output_name": "demo_output_4x.jpg", | |
| "files": { | |
| "demo_input.png": [ | |
| 993, | |
| 747 | |
| ], | |
| "demo_output_4x.jpg": [ | |
| 3972, | |
| 2988 | |
| ], | |
| "demo_compare.jpg": [ | |
| 520, | |
| 278 | |
| ] | |
| } | |
| } | |
| } | |
| } |