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