{
  "model": "NAFNet-GoPro-width32",
  "legacy_file": "NAFNet-GoPro-width32.onnx",
  "legacy_sha256": "4f06e450ddcce675f5766b18c03e755d52eca5124743d5cc648811134777c80a",
  "variants": {
    "fp32": {
      "path": "NAFNet-GoPro-width32_fp32.onnx",
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 69149566,
      "sha256": "535ec94ecd0d1f9c8645854783ff29c72348e58ec27b9458e062325897145380",
      "inputs": [
        {
          "name": "input",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            3,
            "height",
            "width"
          ]
        }
      ],
      "outputs": [
        {
          "name": "output",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            "Sliceoutput_dim_1",
            "output_height",
            "output_width"
          ]
        }
      ]
    },
    "fp16": {
      "path": "NAFNet-GoPro-width32_fp16.onnx",
      "compute": "mixed_fp16_fp32_sensitive_ops",
      "comparisons_to_fp32": [
        [
          {
            "shape": [
              1,
              3,
              64,
              64
            ],
            "mae": 0.010257901491513621,
            "max_abs": 0.0881810188293457,
            "psnr_range_1_db": 37.102926838346846
          }
        ],
        [
          {
            "shape": [
              1,
              3,
              64,
              96
            ],
            "mae": 0.0011675931533419418,
            "max_abs": 0.008857011795043945,
            "psnr_range_1_db": 56.26104410279248
          }
        ],
        [
          {
            "shape": [
              2,
              3,
              64,
              64
            ],
            "mae": 0.00605239109059615,
            "max_abs": 0.0881810188293457,
            "psnr_range_1_db": 39.96990487466185
          }
        ]
      ],
      "runtime": "passed_onnxruntime_cpu",
      "status": "generated",
      "bytes": 35411018,
      "sha256": "bb3f90287c7cf3a8f4e12a4ea59764b44d12f835697700c86ac6819cee0b584f",
      "inputs": [
        {
          "name": "input",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            3,
            "height",
            "width"
          ]
        }
      ],
      "outputs": [
        {
          "name": "output",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            "Sliceoutput_dim_1",
            "output_height",
            "output_width"
          ]
        }
      ]
    },
    "bf16": {
      "path": "NAFNet-GoPro-width32_bf16.onnx",
      "compute": "bf16_conv_matmul_gemm_fp32_other_ops",
      "bf16_compute_nodes": 226,
      "runtime": "requires_native_gpu_validation",
      "status": "generated",
      "bytes": 35062932,
      "sha256": "a0d97e05cf4557921088282e6763b2d085783c732e86e6f5367624094bc2b4c2",
      "inputs": [
        {
          "name": "input",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            3,
            "height",
            "width"
          ]
        }
      ],
      "outputs": [
        {
          "name": "output",
          "dtype": "FLOAT",
          "shape": [
            "batch",
            "Sliceoutput_dim_1",
            "output_height",
            "output_width"
          ]
        }
      ]
    }
  },
  "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance",
  "source": {
    "kind": "original_checkpoint",
    "sha256": "19394e6155d12ef6371d1d57496f87f0ec88f92bdffa27c0792690722d5d1a5c",
    "architecture": "NAFNet",
    "stored_floating_dtypes": [
      "torch.float32"
    ],
    "checkpoint_filename": "NAFNet-GoPro-width32.pth",
    "source_url": "https://drive.usercontent.google.com/download?id=1Fr2QadtDCEXg6iwWX8OzeZLbHOx2t5Bj&export=download&confirm=t"
  },
  "notes": [
    "Fresh PyTorch export from original PTH with dynamic axes; no legacy ONNX graph used."
  ],
  "fp32_runtime_checks": [
    {
      "input": [
        1,
        3,
        64,
        64
      ],
      "outputs": [
        [
          1,
          3,
          64,
          64
        ]
      ],
      "pytorch_comparison": [
        {
          "shape": [
            1,
            3,
            64,
            64
          ],
          "mae": 6.2176402101007016e-06,
          "max_abs": 5.3882598876953125e-05,
          "psnr_range_1_db": 101.50838711408524
        }
      ]
    },
    {
      "input": [
        1,
        3,
        64,
        96
      ],
      "outputs": [
        [
          1,
          3,
          64,
          96
        ]
      ],
      "pytorch_comparison": [
        {
          "shape": [
            1,
            3,
            64,
            96
          ],
          "mae": 5.4098014364475e-07,
          "max_abs": 5.0067901611328125e-06,
          "psnr_range_1_db": 122.87944903401853
        }
      ]
    },
    {
      "input": [
        2,
        3,
        64,
        64
      ],
      "outputs": [
        [
          2,
          3,
          64,
          64
        ]
      ],
      "pytorch_comparison": [
        {
          "shape": [
            2,
            3,
            64,
            64
          ],
          "mae": 2.4163679578729593e-06,
          "max_abs": 3.6716461181640625e-05,
          "psnr_range_1_db": 108.35386831996092
        }
      ]
    }
  ],
  "status": "exported"
}