{ "source_policy": "original_pth_plus_user_authorized_onnx", "model_count": 29, "original_checkpoint_model_count": 23, "onnx_conversion_models": [ "Deblurgan_v2", "face_enhance_x4", "LVAx2", "MSharpx4", "rivagan_decoder", "rivagan_encoder" ], "onnx_file_count": 87, "total_onnx_bytes": 4087623114, "models": [ { "model": "BSRGANx2", "legacy_file": "BSRGANx2_fp16.onnx", "legacy_sha256": "52ed22baec04a8835133286a66fdb5535c88cebda4787799514a20e75cba67cb", "variants": { "fp32": { "path": "BSRGANx2_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 66984173, "sha256": "99d246feec327682ab33f4057c5e3cc89e8861dea3b377f0cc6bb64db673d8a7", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "BSRGANx2_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 128, 128 ], "mae": 0.0007429785570517803, "max_abs": 0.0010609030723571777, "psnr_range_1_db": 62.53733123668034 } ], [ { "shape": [ 1, 3, 128, 192 ], "mae": 0.000759998032461024, "max_abs": 0.0012192726135253906, "psnr_range_1_db": 62.294739251902335 } ], [ { "shape": [ 2, 3, 128, 128 ], "mae": 0.0007484533616661793, "max_abs": 0.0010996460914611816, "psnr_range_1_db": 62.46359754138809 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 33779020, "sha256": "916b2f4b6aa50fd160d82a78b32783ca8ec0f6d30c48400fec58af36da4f3b69", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "BSRGANx2_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 350, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 33707781, "sha256": "f9f79d052675555a322dd1376bebc02ca75d10a3ed8315e66094bd564c0465f2", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 35495540, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 128, 128 ], "mae": 0.0011615341081778752, "max_abs": 0.004794597625732422, "psnr_range_1_db": 56.99784115717941 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 128, 192 ], "mae": 0.0012552296025710853, "max_abs": 0.005989432334899902, "psnr_range_1_db": 56.24299063658352 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 128, 128 ], "mae": 0.0011752913269447163, "max_abs": 0.005030333995819092, "psnr_range_1_db": 56.87212080807111 } ] } ], "seconds": 107.45 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "614f68bcafcf123a1ea51cc75ae5741640a8b1d49dd75757ad3ad531f772c736", "architecture": "ESRGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "BSRGANx2.pth", "source_url": "https://github.com/cszn/KAIR/releases/download/v1.0/BSRGANx2.pth" }, "notes": [], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 128, 128 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 128, 128 ], "mae": 2.1638152247760445e-07, "max_abs": 1.3113021850585938e-06, "psnr_range_1_db": 131.03901564137195 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 128, 192 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 128, 192 ], "mae": 2.2037743797732725e-07, "max_abs": 1.3709068298339844e-06, "psnr_range_1_db": 130.8819716828055 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 128, 128 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 128, 128 ], "mae": 2.1865738138634092e-07, "max_abs": 1.3113021850585938e-06, "psnr_range_1_db": 130.97287463600532 } ] } ], "status": "exported" }, { "model": "BSRGANx4", "legacy_file": "BSRGANx4_fp16.onnx", "legacy_sha256": "ca9bd2e7db018b122dc2e73e5b8d315965491ff3a1bbf983a4764d1ab868b4f2", "variants": { "fp32": { "path": "BSRGANx4_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 67132649, "sha256": "3b53928478d173e8b7604cfe829c21c134bcb9831271e3fc4bc1f1f6f8ee52f1", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "BSRGANx4_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00032134819840360745, "max_abs": 0.001217171549797058, "psnr_range_1_db": 68.56738207150232 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0002953027054759774, "max_abs": 0.0014349594712257385, "psnr_range_1_db": 69.2160042476249 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0003271605085473084, "max_abs": 0.001405566930770874, "psnr_range_1_db": 68.45621019723696 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 33854535, "sha256": "072683a071c4ef600f78de588919228f0231ea2a52a1e2649628f3a3f87bb048", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "BSRGANx4_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 351, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 33782537, "sha256": "037e6f772f3fae73fd3953cdca1800468b2ed00982955a7659abda230fc9b5e3", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 35603780, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0010725154940741295, "max_abs": 0.007605574559420347, "psnr_range_1_db": 57.28329899278222 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.001002010397092887, "max_abs": 0.007382258772850037, "psnr_range_1_db": 57.86607026470033 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0010586708070275581, "max_abs": 0.007605574559420347, "psnr_range_1_db": 57.4969612217454 } ] } ], "seconds": 59.17 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "5d505a0766160921e0388d76e1ddf08cb114303990f9080432bf2b1c988b1c54", "architecture": "ESRGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "BSRGANx4.pth", "source_url": "https://github.com/cszn/KAIR/releases/download/v1.0/BSRGAN.pth" }, "notes": [], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 1.2187293188798708e-07, "max_abs": 9.834766387939453e-07, "psnr_range_1_db": 135.96107804893387 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 1.1755214150708993e-07, "max_abs": 1.1622905731201172e-06, "psnr_range_1_db": 136.28967001979063 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 1.284419475022404e-07, "max_abs": 1.1473894119262695e-06, "psnr_range_1_db": 135.51694617336025 } ] } ], "status": "exported" }, { "model": "codeformer", "legacy_file": "codeformer.onnx", "legacy_sha256": "92dc10cbb67ffff9b5a4ff238fea1721adae0e4981bb6f83ef70236a52d54414", "variants": { "fp32": { "path": "codeformer_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 376884943, "sha256": "49ac5b0a7cba4f0984133574ccbe76b6cc24653c482b6227b323a0129399ef40", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] }, { "name": "w", "dtype": "FLOAT", "shape": [ 1 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] }, "fp16": { "path": "codeformer_fp16.onnx", "converter_fallback_reason": "Keep attention/normalization graph in FP32; convert compute islands directly", "compute": "fp16_conv_matmul_gemm_fp32_other_ops", "fp16_compute_nodes": 215, "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 512, 512 ], "mae": 0.0001628468292148947, "max_abs": 0.003604382276535034, "psnr_range_1_db": 74.08357412730754 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 189135402, "sha256": "10f212bf78faadbe5b48ea5870e623ec2fb34b89854b659222a152d79dad4bf1", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] }, { "name": "w", "dtype": "FLOAT", "shape": [ 1 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] }, "bf16": { "path": "codeformer_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 215, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 189135404, "sha256": "cc5c34e637c574f08e6c54b7f7e5ce37854796cbec062a2a90eaeadc6ff43cda", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] }, { "name": "w", "dtype": "FLOAT", "shape": [ 1 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "1009e537e0c2a07d4cabce6355f53cb66767cd4b4297ec7a4a64ca4b8a5684b7", "architecture": "CodeFormer", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "codeformer.pth", "source_url": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth" }, "notes": [ "Fresh PyTorch export from original PTH with dynamic axes; no legacy ONNX graph used." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 512, 512 ], "outputs": [ [ 1, 3, 512, 512 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 512, 512 ], "mae": 1.850824422392509e-06, "max_abs": 0.00022786855697631836, "psnr_range_1_db": 107.15252302824106 } ] } ], "status": "exported" }, { "model": "codeformer_colorization", "legacy_file": "codeformer_colorization.onnx", "legacy_sha256": "5e8ca901a5582aefbfb727593adc69664937ff7e898f8bf10f7c6f9d4d4c099b", "variants": { "fp32": { "path": "codeformer_colorization_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 334207030, "sha256": "51b60677c455a941b0c37b227a4f15bf802ee4c8c4e99ecbe359b032c3ea1e80", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] }, "fp16": { "path": "codeformer_colorization_fp16.onnx", "converter_fallback_reason": "Keep attention/normalization graph in FP32; convert compute islands directly", "compute": "fp16_conv_matmul_gemm_fp32_other_ops", "fp16_compute_nodes": 187, "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 512, 512 ], "mae": 6.55018518844012e-05, "max_abs": 0.0032164156436920166, "psnr_range_1_db": 79.84506505716755 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 167762869, "sha256": "6b5f06c5c7e3cbde09cdcd693a3a19af75b8987821fd6f45624a788eb126d0a2", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] }, "bf16": { "path": "codeformer_colorization_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 187, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 167762871, "sha256": "e3e82cc5ffd9bc45c6ad8efb17c045a7cad6f6b9b5df912b246236692ee8a48b", "inputs": [ { "name": "x", "dtype": "FLOAT", "shape": [ 1, 3, 512, 512 ] } ], "outputs": [ { "name": "y", "dtype": "FLOAT", "shape": [ "Convy_dim_0", 3, "Convy_dim_0", "Convy_dim_3" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "4f041f194fe17cc128d7c76526fe45689079ce5f5811e1441ccfd3d5aec4693d", "architecture": "CodeFormer", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "codeformer_colorization.pth", "source_url": "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer_colorization.pth" }, "notes": [ "Fresh PyTorch export from original PTH with dynamic axes; no legacy ONNX graph used." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 512, 512 ], "outputs": [ [ 1, 3, 512, 512 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 512, 512 ], "mae": 1.6278586571701e-06, "max_abs": 0.0003707408905029297, "psnr_range_1_db": 103.67533070915515 } ] } ], "status": "exported" }, { "model": "Deblurgan_v2", "legacy_file": "Deblurgan_v2.onnx", "legacy_sha256": "46592ec297e38875399f5ede3ba691daa593036bce9c3853dc65bb91c4801581", "variants": { "fp32": { "path": "Deblurgan_v2_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 144786560, "sha256": "90736fe1509df200be2589be8322a2027d99e087af4e23c47eee1d6e01fa4ac5", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "Deblurgan_v2_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00010846205733313259, "max_abs": 0.0004943609237670898, "psnr_range_1_db": 77.1136714254338 } ], [ { "shape": [ 1, 3, 256, 320 ], "mae": 0.00010849559222, "max_abs": 0.0005135536193847656, "psnr_range_1_db": 77.11609081937137 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00010833534141655814, "max_abs": 0.0005038976669311523, "psnr_range_1_db": 77.1175470477406 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 73181483, "sha256": "38f3eea4f0407aba069e56545698a850d55efaab2c4616b2f2000bb18f801e50", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "current_comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00011086952638169241, "max_abs": 0.0005927085876464844, "psnr_range_1_db": 76.94017012974727 } ], [ { "shape": [ 1, 3, 256, 320 ], "mae": 0.0001111545234382542, "max_abs": 0.0005730390548706055, "psnr_range_1_db": 76.93313782898682 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00011091551497336798, "max_abs": 0.0005927085876464844, "psnr_range_1_db": 76.93778881880222 } ] ] }, "bf16": { "path": "Deblurgan_v2_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 212, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 71761926, "sha256": "1d2b132752da55a2ad5e0e3ceb24ef3e3f1f3bba6eb6a567657110306b12764e", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 73897132, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 256, 256 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0005402937050984491, "max_abs": 0.0030210018157958984, "psnr_range_1_db": 63.20068077330124 } ] }, { "input": [ 1, 3, 256, 320 ], "outputs": [ { "shape": [ 1, 3, 256, 320 ], "mae": 0.0005377690079936789, "max_abs": 0.0030635595321655273, "psnr_range_1_db": 63.25598242816889 } ] }, { "input": [ 2, 3, 256, 256 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0005382559024778762, "max_abs": 0.0037442445755004883, "psnr_range_1_db": 63.23892026000449 } ] } ], "seconds": 34.29 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "legacy_fp32_onnx_graph", "sha256": "46592ec297e38875399f5ede3ba691daa593036bce9c3853dc65bb91c4801581", "onnx_filename": "Deblurgan_v2.onnx" }, "notes": [ "Removed five verified identity Reshapes with frozen spatial dimensions; input H/W must be multiples of 32.", "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 256, 256 ], "outputs": [ [ 1, 3, 256, 256 ] ] }, { "input": [ 1, 3, 256, 320 ], "outputs": [ [ 1, 3, 256, 320 ] ] }, { "input": [ 2, 3, 256, 256 ], "outputs": [ [ 2, 3, 256, 256 ] ] } ], "status": "exported", "validation_evidence": "Identical SHA-256 artifacts retain archived runtime/GPU checks; current CPU and source comparison rechecked.", "onnx_conversion_authorized_by_user": true, "original_onnx_reference_check": { "status": "passed", "source_sha256": "46592ec297e38875399f5ede3ba691daa593036bce9c3853dc65bb91c4801581", "converted_sha256": "90736fe1509df200be2589be8322a2027d99e087af4e23c47eee1d6e01fa4ac5", "precision": "fp32", "input": [ 1, 3, 736, 1280 ], "comparison": [ { "shape": [ 1, 3, 736, 1280 ], "mae": 3.796129315690556e-08, "max_abs": 2.980232238769531e-07, "psnr_range_1_db": 145.47591110546134 } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "model": "DeblurGANv2-FPNInception-GoPro", "legacy_file": null, "legacy_sha256": null, "variants": { "fp32": { "path": "DeblurGANv2-FPNInception-GoPro_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 143965453, "sha256": "9223ca8b70a20a0e5252320cae5efee870ec66145d731bcd7c864935af04a59d", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "DeblurGANv2-FPNInception-GoPro_fp16.onnx", "converter_fallback_reason": "Keep attention/normalization graph in FP32; convert compute islands directly", "compute": "fp16_conv_matmul_gemm_fp32_other_ops", "fp16_compute_nodes": 212, "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00010886180741435474, "max_abs": 0.004001915454864502, "psnr_range_1_db": 75.21770010575607 } ], [ { "shape": [ 1, 3, 256, 320 ], "mae": 0.00045337762773295744, "max_abs": 0.008474737405776978, "psnr_range_1_db": 63.372564548524025 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.002599785258116327, "max_abs": 0.13910765945911407, "psnr_range_1_db": 43.99974850629043 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 72397782, "sha256": "f72fc3faba334bd56f26259ad3bdb2e6ac7f41c01a1a045cc483d23460c468cd", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "DeblurGANv2-FPNInception-GoPro_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 212, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 72397784, "sha256": "34313f053c8ef71c67a15f40962de9a60c9dd921e0b6be09bb6d4f6f2801f331", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 536870912, "tf32": false, "engine_bytes": 76761916, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 256, 256 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0027366119108193456, "max_abs": 0.04217076301574707, "psnr_range_1_db": 48.36914709822048 } ] }, { "input": [ 1, 3, 256, 320 ], "outputs": [ { "shape": [ 1, 3, 256, 320 ], "mae": 0.003968607572705688, "max_abs": 0.060466110706329346, "psnr_range_1_db": 45.00137847804956 } ] }, { "input": [ 2, 3, 256, 256 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.046586357149101344, "max_abs": 1.2938390970230103, "psnr_range_1_db": 20.074337739456457 } ] } ], "seconds": 99.79 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "0ab85a0419534bca6c8bee65eff3c163313e647b4577d493d679aa31169b0eea", "architecture": "DeblurGANv2-FPNInception", "stored_floating_dtypes": [ "torch.float32" ], "source_url": "https://drive.usercontent.google.com/download?id=1UXcsRVW-6KF23_TNzxw-xC0SzaMfXOaR&export=download&confirm=t", "checkpoint_filename": "DeblurGANv2-FPNInception-GoPro.pth" }, "notes": [ "Fresh PyTorch export from original PTH with dynamic axes; no legacy ONNX graph used.", "Official GoPro checkpoint differs from legacy Deblurgan_v2; stored as a separate model.", "Official Predictor uses train(True) statistics. BatchNorm and InstanceNorm use input statistics without state updates, verified against upstream eager inference.", "RGB I/O range [-1,1]; H/W multiples of 32, tested from 256 pixels. External preprocessing handles padding/cropping." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 256, 256 ], "outputs": [ [ 1, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 3.0931137125674485e-07, "max_abs": 8.344650268554688e-06, "psnr_range_1_db": 125.96752025165861 } ] }, { "input": [ 1, 3, 256, 320 ], "outputs": [ [ 1, 3, 256, 320 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 320 ], "mae": 2.3379352095768506e-07, "max_abs": 1.150369644165039e-05, "psnr_range_1_db": 126.59254674157404 } ] }, { "input": [ 2, 3, 256, 256 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 5.039054577493651e-06, "max_abs": 0.00030535459518432617, "psnr_range_1_db": 98.18143002521848 } ] } ], "status": "exported" }, { "model": "face_enhance_x4", "legacy_file": "face_enhance_x4.onnx", "legacy_sha256": "cf972ae320db2445d47e991c63f7c070f093218926a99cb5a476a41b197cc587", "variants": { "fp32": { "path": "face_enhance_x4_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 66160201, "sha256": "c6401eede1aedcf9384ff0d1bc42298560504d606ca074d81eb104860de86716", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "face_enhance_x4_fp16.onnx", "converter_fallback_reason": "Keep attention/normalization graph in FP32; convert compute islands directly", "compute": "fp16_conv_matmul_gemm_fp32_other_ops", "fp16_compute_nodes": 624, "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00010680464364061966, "max_abs": 0.0007883310317993164, "psnr_range_1_db": 77.50716927293814 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.00010937072678037414, "max_abs": 0.0007890760898590088, "psnr_range_1_db": 77.30832198205319 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00011271557355030382, "max_abs": 0.0007820725440979004, "psnr_range_1_db": 77.15372318511298 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 36926874, "sha256": "eb7df4de6690fb224a288584f8cf05d5fb1bef07d09e97de5b9fd60be68f54b6", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "current_comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00010680464364061966, "max_abs": 0.0007883310317993164, "psnr_range_1_db": 77.50716927293814 } ], [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00011088519600485192, "max_abs": 0.00078544020652771, "psnr_range_1_db": 77.17037861271258 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00011271557355030382, "max_abs": 0.0007820725440979004, "psnr_range_1_db": 77.15372318511298 } ] ] }, "bf16": { "path": "face_enhance_x4_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 624, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 36928226, "sha256": "e0f0c8badc87f82a41750548765ab51d1c9c9d88df5f46f487a0b12e0a2c6f22", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "status": "not_completed", "reason": "Native BF16 engine build exceeded several minutes and was stopped during compilation; full ONNX type/shape checks and FP32/FP16 CPU inference passed." } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "legacy_fp32_onnx_graph", "sha256": "cf972ae320db2445d47e991c63f7c070f093218926a99cb5a476a41b197cc587", "onnx_filename": "face_enhance_x4.onnx" }, "notes": [ "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 256, 256 ] ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ [ 1, 3, 320, 384 ] ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ] } ], "status": "exported", "validation_evidence": "Identical SHA-256 artifacts retain archived runtime/GPU checks; current CPU and source comparison rechecked.", "onnx_conversion_authorized_by_user": true, "original_onnx_reference_check": { "status": "passed", "source_sha256": "cf972ae320db2445d47e991c63f7c070f093218926a99cb5a476a41b197cc587", "converted_sha256": "c6401eede1aedcf9384ff0d1bc42298560504d606ca074d81eb104860de86716", "precision": "fp32", "input": [ 1, 3, 64, 64 ], "comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0, "max_abs": 0.0, "psnr_range_1_db": null } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "model": "IRCNN_Lx1", "legacy_file": "IRCNN_Lx1_fp16.onnx", "legacy_sha256": "2e1dd14ebab42c230d2879a2bde2f6687912eb27f2e8f4052e7aefc944a8fe1e", "variants": { "fp32": { "path": "IRCNN_Lx1_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 755288, "sha256": "fec49805d8a8e07d0bb3f616f572cd755ff136f4ee45e753484c8de13cd664d3", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "IRCNN_Lx1_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 6.293406204349594e-05, "max_abs": 0.00034368038177490234, "psnr_range_1_db": 82.07293759812556 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 6.186403321508906e-05, "max_abs": 0.0003171563148498535, "psnr_range_1_db": 82.22702766284804 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 6.298989319475368e-05, "max_abs": 0.00034368038177490234, "psnr_range_1_db": 82.06663858977339 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 380421, "sha256": "5c5b05c4060612e06ca4df134f2bd0d49330a1b827aa0aac674d439d0b65dd02", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "IRCNN_Lx1_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 7, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 380003, "sha256": "0ad9c5c3e0c9ae76a6fb03f4f92b41b5203239fefea37f63518199518afaf02b", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 523044, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 64, 64 ], "mae": 0.0006665728553040632, "max_abs": 0.0033580362796783447, "psnr_range_1_db": 61.58878350374113 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 64, 96 ], "mae": 0.0006474270865560862, "max_abs": 0.0034367740154266357, "psnr_range_1_db": 61.81013111919985 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 64, 64 ], "mae": 0.0006617188832175694, "max_abs": 0.003744959831237793, "psnr_range_1_db": 61.647001637544356 } ] } ], "seconds": 47.46 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "6b54ef8d8797852f461f68f871bfb1cea63ee2418f68fcec79d6e818d952994e", "architecture": "IRCNN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "IRCNN_Lx1.pth" }, "notes": [], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 64 ], "mae": 9.797630203441561e-08, "max_abs": 5.364418029785156e-07, "psnr_range_1_db": 138.0720997496727 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 9.742936729000778e-08, "max_abs": 6.258487701416016e-07, "psnr_range_1_db": 138.0857751265174 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 9.85109333366078e-08, "max_abs": 5.364418029785156e-07, "psnr_range_1_db": 138.0197293584494 } ] } ], "status": "exported" }, { "model": "IRCNN_Mx1", "legacy_file": "IRCNN_Mx1_fp16.onnx", "legacy_sha256": "9675737b5c999c17d5d5480bc453a881ed12c59781c31ef3d9fcfb2d2ef26d9b", "variants": { "fp32": { "path": "IRCNN_Mx1_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 755288, "sha256": "a81151814e04cb2ef7d9e45510f8c905d90ff0d254399aae3b4dc373fb5b731f", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "IRCNN_Mx1_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 4.8752062942488315e-05, "max_abs": 0.00024402141571044922, "psnr_range_1_db": 84.20136472359046 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 4.79250522099998e-05, "max_abs": 0.00023889541625976562, "psnr_range_1_db": 84.32017096059312 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 4.8277318285272486e-05, "max_abs": 0.00024402141571044922, "psnr_range_1_db": 84.27446138311504 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 380421, "sha256": "9b0b15e9b4a9b665f8e45bfc78d0ad0a1018e24a6d0e5ff1ee92acae1a322101", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "IRCNN_Mx1_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 7, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 380003, "sha256": "92f7fbd5cc596b715dbedd31c7c8ecc27d0a4822a6df948b420d44ed92319945", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 523044, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 64, 64 ], "mae": 0.00044655248962044425, "max_abs": 0.0023169368505477905, "psnr_range_1_db": 64.98610804098394 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 64, 96 ], "mae": 0.00044668022443753824, "max_abs": 0.0025789737701416016, "psnr_range_1_db": 64.97661190228241 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 64, 64 ], "mae": 0.0004460422795583933, "max_abs": 0.0023700445890426636, "psnr_range_1_db": 64.9910121910939 } ] } ], "seconds": 2.19 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "88a6f5d1cb741e4a66637fe2b111c2b93a5242d712f2963e0fb4b85eb11f0255", "architecture": "IRCNN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "IRCNN_Mx1.pth" }, "notes": [], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 64 ], "mae": 6.178843629337886e-08, "max_abs": 3.0547380447387695e-07, "psnr_range_1_db": 141.9100885705353 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 6.19946705329413e-08, "max_abs": 3.5762786865234375e-07, "psnr_range_1_db": 141.90332899223418 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 6.22010721447926e-08, "max_abs": 3.5762786865234375e-07, "psnr_range_1_db": 141.83116856735603 } ] } ], "status": "exported" }, { "model": "LVAx2", "legacy_file": "LVAx2_fp16.onnx", "legacy_sha256": "3ae7780f6f5d3695b2bbac62c0c54b90545e29f9bddca5317ec642296d275378", "status": "exported", "source": { "kind": "legacy_fp16_onnx_graph", "sha256": "3ae7780f6f5d3695b2bbac62c0c54b90545e29f9bddca5317ec642296d275378", "stored_floating_dtypes": [ "FLOAT16" ], "onnx_filename": "LVAx2_fp16.onnx" }, "variants": { "fp32": { "path": "LVAx2_fp32.onnx", "status": "generated", "bytes": 1659300, "sha256": "b9e77ab50b7a87938e821b5f7afa7dd1a68819c2b0ad9f3915c1b2080f4075f3", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "runtime": "passed_onnxruntime_cpu", "precision_semantics": "fp32_execution_promoted_fp16_weights" }, "fp16": { "path": "LVAx2_fp16.onnx", "status": "generated", "bytes": 839909, "sha256": "9d3bee3a6a7cee389d3661b65a4f259e75ec353bbaff9b16a0acd32db567832e", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "runtime": "passed_onnxruntime_cpu", "current_comparisons_to_fp32": [ [ { "shape": [ 1, 3, 128, 128 ], "mae": 7.817742418107325e-05, "max_abs": 0.000244140625, "psnr_range_1_db": 80.02921503667548 } ], [ { "shape": [ 1, 3, 160, 192 ], "mae": 7.866326272455303e-05, "max_abs": 0.000244140625, "psnr_range_1_db": 79.99381254025931 } ], [ { "shape": [ 2, 3, 128, 128 ], "mae": 7.804819426837639e-05, "max_abs": 0.000244140625, "psnr_range_1_db": 80.04041940282605 } ] ] }, "bf16": { "path": "LVAx2_bf16.onnx", "status": "generated", "bytes": 834731, "sha256": "c4cb5db4f9808e23139664cf504dbe8d2a8369290e56a2408cf9b74cd29497f6", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ 1, 3, "output_height", "output_width" ] } ], "runtime": "passed_tensorrt_gpu", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 22, "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 536870912, "tf32": false, "engine_bytes": 1275268, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 128, 128 ], "mae": 0.0023082519174598324, "max_abs": 0.022345900535583496, "psnr_range_1_db": 50.457839756605 } ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ { "shape": [ 1, 3, 160, 192 ], "mae": 0.002345463626301757, "max_abs": 0.034210026264190674, "psnr_range_1_db": 50.2505654724863 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 128, 128 ], "mae": 0.002302888141950158, "max_abs": 0.02291262149810791, "psnr_range_1_db": 50.43593511939662 } ] } ], "seconds": 70.35 } } }, "low_bit_quantization": "not_requested_for_light_models", "notes": [ "FP32 execution uses promoted FP16 source weights; original FP32 precision cannot be recovered.", "FP16 keeps original model arithmetic; only dynamic batch/channel metadata is corrected.", "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "onnx_conversion_authorized_by_user": true, "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 128, 128 ] ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ [ 1, 3, 160, 192 ] ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 128, 128 ] ] } ], "original_onnx_reference_check": { "status": "passed", "source_sha256": "3ae7780f6f5d3695b2bbac62c0c54b90545e29f9bddca5317ec642296d275378", "converted_sha256": "9d3bee3a6a7cee389d3661b65a4f259e75ec353bbaff9b16a0acd32db567832e", "precision": "fp16", "input": [ 1, 3, 64, 64 ], "comparison": [ { "shape": [ 1, 3, 128, 128 ], "mae": 0.0, "max_abs": 0.0, "psnr_range_1_db": null } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "model": "MSharpx4", "legacy_file": "MSharpx4_fp16.onnx", "legacy_sha256": "f8caf7e25ba3c5f37354e0185fe860bf336ae8a9cfdf745b864d8f4da10279cb", "status": "exported", "source": { "kind": "legacy_fp16_onnx_graph", "sha256": "f8caf7e25ba3c5f37354e0185fe860bf336ae8a9cfdf745b864d8f4da10279cb", "stored_floating_dtypes": [ "FLOAT16" ], "onnx_filename": "MSharpx4_fp16.onnx" }, "variants": { "fp32": { "path": "MSharpx4_fp32.onnx", "status": "generated", "bytes": 30014564, "sha256": "7c90fbcbc33f07844b7075400b202211138639c1dccff2cf199e8f9e3eac007f", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "runtime": "passed_onnxruntime_cpu", "precision_semantics": "fp32_execution_promoted_fp16_weights" }, "fp16": { "path": "MSharpx4_fp16.onnx", "status": "generated", "bytes": 15275730, "sha256": "8d209d1bb09fe7b7a1f6134d4bc98343fba723fb35d53ddb2258d00bc533f736", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "runtime": "passed_onnxruntime_cpu", "current_comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 9.241798814703846e-05, "max_abs": 0.00047516822814941406, "psnr_range_1_db": 78.62339373938107 } ], [ { "shape": [ 1, 3, 320, 384 ], "mae": 9.23772332902849e-05, "max_abs": 0.0005388855934143066, "psnr_range_1_db": 78.61247667424611 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 9.194720490768304e-05, "max_abs": 0.0004890859127044678, "psnr_range_1_db": 78.64596698354042 } ] ] }, "bf16": { "path": "MSharpx4_bf16.onnx", "status": "generated", "bytes": 15124068, "sha256": "1f2af498119acfbc6fad8373ad4371e1da2569a2d7d79fe143fb221396aec9f3", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ 1, 3, "output_height", "output_width" ] } ], "runtime": "passed_tensorrt_gpu", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 142, "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 536870912, "tf32": false, "engine_bytes": 17767188, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0012639994887801247, "max_abs": 0.014169692993164062, "psnr_range_1_db": 55.92430167003972 } ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.0012765380073081662, "max_abs": 0.010266900062561035, "psnr_range_1_db": 55.81575843320759 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0012466330885179862, "max_abs": 0.014169692993164062, "psnr_range_1_db": 56.05267313701162 } ] } ], "seconds": 37.56 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "notes": [ "FP32 execution uses promoted FP16 source weights; original FP32 precision cannot be recovered.", "FP16 keeps original model arithmetic; only dynamic batch/channel metadata is corrected.", "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "onnx_conversion_authorized_by_user": true, "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 256, 256 ] ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ [ 1, 3, 320, 384 ] ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ] } ], "original_onnx_reference_check": { "status": "passed", "source_sha256": "f8caf7e25ba3c5f37354e0185fe860bf336ae8a9cfdf745b864d8f4da10279cb", "converted_sha256": "8d209d1bb09fe7b7a1f6134d4bc98343fba723fb35d53ddb2258d00bc533f736", "precision": "fp16", "input": [ 1, 3, 64, 64 ], "comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0, "max_abs": 0.0, "psnr_range_1_db": null } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "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" }, { "model": "Real-CUGAN_v3_2x", "legacy_file": "Real-CUGAN_v3_2x.onnx", "legacy_sha256": "3066bd5f13052b7f69741076029c40e7e5c74d82ffc849c55a96e4cbbe161d82", "variants": { "fp32": { "path": "Real-CUGAN_v3_2x_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 5172789, "sha256": "94a33ba26f17fd9b3b78930aa3da40040abdb3ea44d2e402e0a640683d8624cc", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "fp16": { "path": "Real-CUGAN_v3_2x_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 128, 128 ], "mae": 0.04859780566766858, "max_abs": 0.187591552734375, "psnr_range_1_db": 24.62947809087316 } ], [ { "shape": [ 1, 3, 128, 192 ], "mae": 0.04694390597028865, "max_abs": 0.1565093994140625, "psnr_range_1_db": 24.91435672934117 } ], [ { "shape": [ 2, 3, 128, 128 ], "mae": 0.04957877479804059, "max_abs": 0.187591552734375, "psnr_range_1_db": 24.498188599041146 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2619303, "sha256": "32c1cc950818b83176957fa3b41f8d62a9620f1bd664c80b412260eca6ea3db4", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "bf16": { "path": "Real-CUGAN_v3_2x_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 26, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 2810944, "sha256": "2f1220408cd83be051085e4bc310b195bcd1f95e7069a545416637c4e0e464d8", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "0a14739f3f5fcbd74ec3ce2806d13a47916c916b20afe4a39d95f6df4ca6abd8", "architecture": "RealCUGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Real-CUGAN_v3_2x.pth" }, "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, 128, 128 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 128, 128 ], "mae": 6.549013778567314e-05, "max_abs": 0.0003662109375, "psnr_range_1_db": 81.60432591855016 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 128, 192 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 128, 192 ], "mae": 6.582753525839912e-05, "max_abs": 0.00042724609375, "psnr_range_1_db": 81.56710587497437 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 128, 128 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 128, 128 ], "mae": 6.555261400838693e-05, "max_abs": 0.000396728515625, "psnr_range_1_db": 81.59185898604332 } ] } ], "status": "exported" }, { "model": "Real-CUGAN_v3_3x", "legacy_file": "Real-CUGAN_v3_3x.onnx", "legacy_sha256": "5611941eb1203dc25be4462f86b38228ab1057c8c8784d57c0c24277d6259efd", "variants": { "fp32": { "path": "Real-CUGAN_v3_3x_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 5179703, "sha256": "e8c7e006f178b82bf408852a225d99d7ea0165867b5fd098f4ff7abd8e4cbb08", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "fp16": { "path": "Real-CUGAN_v3_3x_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 192, 192 ], "mae": 0.0783417226953639, "max_abs": 0.542755126953125, "psnr_range_1_db": 20.488779694114584 } ], [ { "shape": [ 1, 3, 192, 288 ], "mae": 0.08303776673144764, "max_abs": 0.6135406494140625, "psnr_range_1_db": 20.044201000430416 } ], [ { "shape": [ 2, 3, 192, 192 ], "mae": 0.08066301085744743, "max_abs": 0.542755126953125, "psnr_range_1_db": 20.358680602580755 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2622759, "sha256": "2254059ae7b91e7e6f6a94258c4ad4ed9bc5f9957521fec486d0a179d554f27c", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "bf16": { "path": "Real-CUGAN_v3_3x_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 26, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 2817858, "sha256": "a3209dfec76aaa38a254e5be4165241c324d30adbdf7dfe32d3d9dfbd0f45d51", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "39f1e6e90d50e5528a63f4ba1866bad23365a737cbea22a80769b2ec4c1c3285", "architecture": "RealCUGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Real-CUGAN_v3_3x.pth" }, "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, 192, 192 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 192, 192 ], "mae": 9.173699827105911e-05, "max_abs": 0.000762939453125, "psnr_range_1_db": 78.67395276586714 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 192, 288 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 192, 288 ], "mae": 8.494464427600672e-05, "max_abs": 0.00079345703125, "psnr_range_1_db": 79.38897148082718 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 192, 192 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 192, 192 ], "mae": 8.018373477238196e-05, "max_abs": 0.0005950927734375, "psnr_range_1_db": 79.92204791150284 } ] } ], "status": "exported" }, { "model": "Real-CUGAN_v3_4x", "legacy_file": "Real-CUGAN_v3_4x.onnx", "legacy_sha256": "a2b07c833e4e3b0b0f2f15a5e0f0ffdf4d3e01a8a8d1892d3fd41375adac5f5e", "variants": { "fp32": { "path": "Real-CUGAN_v3_4x_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 5664430, "sha256": "06b9def7bc2a514d22949e1fc764e34a64b73ad7854cf09a446c20fdb823264f", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "fp16": { "path": "Real-CUGAN_v3_4x_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.09377591391845878, "max_abs": 1.2317352294921875, "psnr_range_1_db": 18.23639152249762 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0931518535517676, "max_abs": 0.964691162109375, "psnr_range_1_db": 18.331447717997012 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.09142321595618341, "max_abs": 1.2317352294921875, "psnr_range_1_db": 18.478685144090697 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2868151, "sha256": "dec468ce07fb1b7b103e889629b4464502713ad7a068539957db98b7f5f867e7", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] }, "bf16": { "path": "Real-CUGAN_v3_4x_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 27, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 3183363, "sha256": "a3f5f0b2861daf7a249ca2d2f60d9c185e1f426b16ef275be53290a309ef2ca4", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", "Muloutput_dim_1", "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "42bd8fcdae37c12c5b25ed59625266bfa65780071a8d38192d83756cb85e98dd", "architecture": "RealCUGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Real-CUGAN_v3_4x.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00013671781440886358, "max_abs": 0.0014190673828125, "psnr_range_1_db": 75.0512086731372 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 8.872862215664807e-05, "max_abs": 0.0012664794921875, "psnr_range_1_db": 78.60600467295089 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0001365069509423241, "max_abs": 0.0018157958984375, "psnr_range_1_db": 75.0090087373423 } ] } ], "status": "exported" }, { "model": "Real_ESRGAN_x4", "legacy_file": "Real_ESRGAN_x4.onnx", "legacy_sha256": "dd8df3e9c800eb9ab75d19398904e5ba818852a1eaad7e70eaa25acf38dc8576", "variants": { "fp32": { "path": "Real_ESRGAN_x4_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 67132649, "sha256": "1af16d764fb1506f7629ebcc060c0f997dde9672ebad33d6c5c3b10ab1eff46b", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "Real_ESRGAN_x4_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00020908163719468575, "max_abs": 0.003976702690124512, "psnr_range_1_db": 71.92296852324526 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.00019381982302648894, "max_abs": 0.0012119710445404053, "psnr_range_1_db": 72.7668016897817 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00020730084221061892, "max_abs": 0.003976702690124512, "psnr_range_1_db": 72.0529926589919 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 33854535, "sha256": "a10914af43a04a583ae8e6db093cd31f295183b91088adf95df475e8a8faaffa", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "Real_ESRGAN_x4_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 351, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 33791949, "sha256": "d2f13fcb8eddbb122f5cd1f95f59a78e8b3b1db48388f5a921a7d651d5a95d15", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1", "architecture": "ESRGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Real_ESRGAN_x4.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 3.0473529477603734e-07, "max_abs": 4.276633262634277e-06, "psnr_range_1_db": 128.14598241097192 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 2.9507999796704905e-07, "max_abs": 2.0265579223632812e-06, "psnr_range_1_db": 128.500544539578 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 2.9952479962958023e-07, "max_abs": 3.7550926208496094e-06, "psnr_range_1_db": 128.33599032113443 } ] } ], "status": "exported" }, { "model": "Real_ESRGAN_x4plus", "legacy_file": "Real_ESRGAN_x4plus.onnx", "legacy_sha256": "0f89575f86c6f3e7321fbb4fdf21732d4324ba81823d12d0271ae381341155ff", "variants": { "fp32": { "path": "Real_ESRGAN_x4plus_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 67132649, "sha256": "1af16d764fb1506f7629ebcc060c0f997dde9672ebad33d6c5c3b10ab1eff46b", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "Real_ESRGAN_x4plus_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00020908163719468575, "max_abs": 0.003976702690124512, "psnr_range_1_db": 71.92296852324526 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.00019381982302648894, "max_abs": 0.0012119710445404053, "psnr_range_1_db": 72.7668016897817 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00020730084221061892, "max_abs": 0.003976702690124512, "psnr_range_1_db": 72.0529926589919 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 33854535, "sha256": "a10914af43a04a583ae8e6db093cd31f295183b91088adf95df475e8a8faaffa", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "Real_ESRGAN_x4plus_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 351, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 33782537, "sha256": "113d47dffb57f39792a05c2e403e92fd19f98d88d40e8ca595ff358e0c24345c", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 35603780, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0011652096141763952, "max_abs": 0.02602139115333557, "psnr_range_1_db": 56.54041306555661 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0011327205969185645, "max_abs": 0.0074385106563568115, "psnr_range_1_db": 57.01994264273502 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0011414742899660268, "max_abs": 0.02602139115333557, "psnr_range_1_db": 56.81641019176139 } ] } ], "seconds": 62.63 } } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "4fa0d38905f75ac06eb49a7951b426670021be3018265fd191d2125df9d682f1", "architecture": "ESRGAN", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Real_ESRGAN_x4plus.pth" }, "notes": [], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 3.0473529477603734e-07, "max_abs": 4.276633262634277e-06, "psnr_range_1_db": 128.14598241097192 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 2.9507999796704905e-07, "max_abs": 2.0265579223632812e-06, "psnr_range_1_db": 128.500544539578 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 2.9952479962958023e-07, "max_abs": 3.7550926208496094e-06, "psnr_range_1_db": 128.33599032113443 } ] } ], "status": "exported" }, { "model": "RealESR_Animex4", "legacy_file": "RealESR_Animex4_fp16.onnx", "legacy_sha256": "6e269ea53fb1835e70532327e10e994ed42446ab605e45632a52807a6a181e72", "variants": { "fp32": { "path": "RealESR_Animex4_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2492916, "sha256": "134b51d33d448a07e8cbe8207ac1d20a1af2a0043962f0afcac64ee29819f465", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "RealESR_Animex4_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0002471529477029814, "max_abs": 0.00204312801361084, "psnr_range_1_db": 70.00856613834648 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.00024637149364379083, "max_abs": 0.0017707645893096924, "psnr_range_1_db": 70.08230170187096 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00024694276814329896, "max_abs": 0.00204312801361084, "psnr_range_1_db": 70.02745176516096 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 1253375, "sha256": "b44b1e57a7b86eff7f2dcd2ab3dae308f8eb1632acf88a763e2d6e039f99da56", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "RealESR_Animex4_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 18, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 1254954, "sha256": "d280a780a6f21ecb5ad4b75d037c9a86cddeb09afc7df1ad20dbd0d846cc302f", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 1450284, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.002762390540264429, "max_abs": 0.023199468851089478, "psnr_range_1_db": 48.971033348986374 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0027845546705628343, "max_abs": 0.023693367838859558, "psnr_range_1_db": 48.891689302570775 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0027590552636098664, "max_abs": 0.023882746696472168, "psnr_range_1_db": 48.969882792407844 } ] } ], "seconds": 99.67 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "b8a8376811077954d82ca3fcf476f1ac3da3e8a68a4f4d71363008000a18b75d", "architecture": "Compact", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "RealESR_Animex4.pth", "source_url": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-animevideov3.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 4.476999440801895e-07, "max_abs": 3.3974647521972656e-06, "psnr_range_1_db": 124.69853739492447 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 4.427639536983305e-07, "max_abs": 4.0531158447265625e-06, "psnr_range_1_db": 124.80656540755804 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 4.419511762480492e-07, "max_abs": 3.516674041748047e-06, "psnr_range_1_db": 124.82584416164529 } ] } ], "status": "exported" }, { "model": "RealESR_Animex4_v2", "legacy_file": null, "legacy_sha256": null, "variants": { "fp32": { "path": "RealESR_Animex4_v2_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2492916, "sha256": "09a95bf7db900fa461cad912cf014a85a71dfd40c0bd720237e974f029910128", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "RealESR_Animex4_v2_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0003348372187090831, "max_abs": 0.002690911293029785, "psnr_range_1_db": 67.37561275463275 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0003335317571932074, "max_abs": 0.002534210681915283, "psnr_range_1_db": 67.40845657132414 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0003329306661470355, "max_abs": 0.002690911293029785, "psnr_range_1_db": 67.41355825493132 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 1253375, "sha256": "5bd07a5a2ac1c91a1d5b4f1f8feac58eb135d0c654c45829fbe4828ad3331ac6", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "RealESR_Animex4_v2_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 18, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 1254954, "sha256": "0fef0f907900bd43e8680ec15f1e1c0da5eb9239e99ca1658c731e3a0813fa1e", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 1450284, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0033273982091159837, "max_abs": 0.027790546417236328, "psnr_range_1_db": 47.34519797527034 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.003310590898724936, "max_abs": 0.02972126007080078, "psnr_range_1_db": 47.39375711790143 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.003316443604212793, "max_abs": 0.030151784420013428, "psnr_range_1_db": 47.37637200050334 } ] } ], "seconds": 7.35 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "5e04b111c16d5b40deb8f8062d09fdff72fa80281fb9607be23b62650c08daff", "architecture": "Compact", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "RealESR_Animex4_v2.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 5.610663068959335e-07, "max_abs": 4.559755325317383e-06, "psnr_range_1_db": 122.77150257992373 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 5.52940054237499e-07, "max_abs": 5.0067901611328125e-06, "psnr_range_1_db": 122.89602932597391 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 5.597844232596572e-07, "max_abs": 5.140900611877441e-06, "psnr_range_1_db": 122.80478618868176 } ] } ], "status": "exported" }, { "model": "RealESR_Gx4", "legacy_file": "RealESR_Gx4_fp16.onnx", "legacy_sha256": "c0a73707e1f487c38ca62607fe4193828c1f5f1af682e844d3d880bc066d6686", "variants": { "fp32": { "path": "RealESR_Gx4_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 4866436, "sha256": "eb41c355ee71e4583fed3d2171c4bd7d3d2ebe59acf4fc064d9795f7415f3c6a", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "RealESR_Gx4_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.00027696707017336547, "max_abs": 0.0019773244857788086, "psnr_range_1_db": 68.72721293137678 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0002785477432275406, "max_abs": 0.0020246803760528564, "psnr_range_1_db": 68.67089244686545 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.00027842761154298995, "max_abs": 0.0022069215774536133, "psnr_range_1_db": 68.70039593529286 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2445376, "sha256": "e7f42d9909623af9cf95e86705a84b726eb5f28b0791a8d08d82937c93035838", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "RealESR_Gx4_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 34, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 2449229, "sha256": "79967f8b1eb49498b7542e18eba2926dadbc1ec50105476bb7357af8393ea87d", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 2689532, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.002770280764451627, "max_abs": 0.02213221788406372, "psnr_range_1_db": 48.86657745384824 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0028096038907960974, "max_abs": 0.025955259799957275, "psnr_range_1_db": 48.77453714529048 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0028143648320574735, "max_abs": 0.024355769157409668, "psnr_range_1_db": 48.74359141988428 } ] } ], "seconds": 10.32 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "8dc7edb9ac80ccdc30c3a5dca6616509367f05fbc184ad95b731f05bece96292", "architecture": "Compact", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "RealESR_Gx4.pth", "source_url": "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 4.865586333835381e-07, "max_abs": 4.649162292480469e-06, "psnr_range_1_db": 123.96532371935764 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 4.887713329379848e-07, "max_abs": 4.76837158203125e-06, "psnr_range_1_db": 123.90458342136296 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 4.909201270682692e-07, "max_abs": 4.112720489501953e-06, "psnr_range_1_db": 123.85847384207383 } ] } ], "status": "exported" }, { "model": "RealESR_Gx4_WDN", "legacy_file": null, "legacy_sha256": null, "variants": { "fp32": { "path": "RealESR_Gx4_WDN_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 4866436, "sha256": "73850ec66b6dfcefe2c6453f31caad92dbeca8411d4ddb5c2695db363221a90b", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "RealESR_Gx4_WDN_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.0006484372807804561, "max_abs": 0.007368266582489014, "psnr_range_1_db": 61.02090777622422 } ], [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.0006472206988108711, "max_abs": 0.014560341835021973, "psnr_range_1_db": 61.00614359908591 } ], [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.0006514159067260531, "max_abs": 0.009568482637405396, "psnr_range_1_db": 60.90832769509908 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 2445376, "sha256": "61851c048f709ac891641c2736a675458a610214068d3c9739e01070a3689a47", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "RealESR_Gx4_WDN_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 34, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 2449229, "sha256": "19dbf388aaf8a58c9bce42b9130ae5e4192eadfeffa22f39d75c3f9669a559fd", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 2689532, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 256, 256 ], "mae": 0.007996666038638503, "max_abs": 0.08836773037910461, "psnr_range_1_db": 39.31567672649475 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 256, 384 ], "mae": 0.008268513306142545, "max_abs": 0.12560230493545532, "psnr_range_1_db": 38.99636780960326 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 256, 256 ], "mae": 0.008176046578493393, "max_abs": 0.11208987236022949, "psnr_range_1_db": 39.07157259689419 } ] } ], "seconds": 8.33 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "1641f8c4464b9f097c9fdda5589273713f67cf59f3d909e0bd688f0cee269dca", "architecture": "Compact", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "RealESR_Gx4_WDN.pth" }, "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, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 256 ], "mae": 1.2103721047651561e-06, "max_abs": 1.5616416931152344e-05, "psnr_range_1_db": 115.61082929594086 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 256, 384 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 256, 384 ], "mae": 1.2278298824084131e-06, "max_abs": 1.728534698486328e-05, "psnr_range_1_db": 115.51865241763623 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 256, 256 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 256, 256 ], "mae": 1.2024798348164722e-06, "max_abs": 1.6868114471435547e-05, "psnr_range_1_db": 115.64123995356672 } ] } ], "status": "exported" }, { "model": "Restormer-Defocus", "legacy_file": "Restormer-Defocus.onnx", "legacy_sha256": "20b532f93bb57297cc8f87731e991058372e4185d02f31b1964dba9c4e4aea36", "variants": { "fp32": { "path": "Restormer-Defocus_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 107167733, "sha256": "a67b476ad969a41a537f2a4308093600deee56efb0986f27ffc217d69798bde0", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "Restormer-Defocus_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 8.721068252270925e-05, "max_abs": 0.0005040764808654785, "psnr_range_1_db": 79.18109510695939 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 8.789375786161852e-05, "max_abs": 0.0004913210868835449, "psnr_range_1_db": 79.09687114193295 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 8.847314249275466e-05, "max_abs": 0.0005044341087341309, "psnr_range_1_db": 79.04298652714839 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 56422808, "sha256": "b52ba4012d96494f7f422bd34ecd37a4fa38184e35f859c1e5f99b994429e00e", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "Restormer-Defocus_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 362, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 55035105, "sha256": "2d39c220ee5fd476da80055217aa2511e25e685fdc90ffccae3833b25c90eea1", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "7dce451f33f8f5e0faf7c4e3996e5dcc1bd425ecd1ada99b0f9750e490fd4c9e", "architecture": "Restormer", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Restormer-Defocus.pth", "source_url": "https://github.com/swz30/Restormer/releases/download/v1.0/single_image_defocus_deblurring.pth" }, "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": 7.862824228747438e-08, "max_abs": 5.662441253662109e-07, "psnr_range_1_db": 139.69233535443442 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 8.087216782668192e-08, "max_abs": 4.76837158203125e-07, "psnr_range_1_db": 139.4978758848223 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 8.024941659338462e-08, "max_abs": 5.960464477539062e-07, "psnr_range_1_db": 139.56096470119854 } ] } ], "status": "exported" }, { "model": "Restormer-Motion", "legacy_file": "Restormer-Motion.onnx", "legacy_sha256": "5dbcf95ac987173b40b13b349bc1f1cbd546c0149651514c1c46e1c025ab09ee", "variants": { "fp32": { "path": "Restormer-Motion_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 107167733, "sha256": "2a2ecfe14be281ff18fb8dfdbe010381386b13b7c8bde779f2583ea92f34412c", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "Restormer-Motion_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 0.0001221492449682652, "max_abs": 0.0006929636001586914, "psnr_range_1_db": 76.00985559808987 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 0.00012506460153518774, "max_abs": 0.0008702874183654785, "psnr_range_1_db": 75.74317353589974 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 0.00012228410743849358, "max_abs": 0.0007673501968383789, "psnr_range_1_db": 76.06045564440751 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 56422808, "sha256": "c139b2cf24d426ad35a63e176e7b8cc5c210ff1c05dd0d8cf88b2dd926703c04", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "Restormer-Motion_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 362, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 55035105, "sha256": "84fefd3b5e4736a0e9601093c20eeceefe13f43f395435eea49a30e60c960cfe", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "194e38fb5b607c9dc5a5b3e08e65b2e79ee2bf0ef5048e0612f6b2ff2f79da31", "architecture": "Restormer", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "Restormer-Motion.pth", "source_url": "https://github.com/swz30/Restormer/releases/download/v1.0/motion_deblurring.pth" }, "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": 1.263523519886197e-07, "max_abs": 1.3113021850585938e-06, "psnr_range_1_db": 135.2481095415686 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 1.4114246798473889e-07, "max_abs": 1.0132789611816406e-06, "psnr_range_1_db": 134.53424900683356 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 1.261835234818894e-07, "max_abs": 1.5497207641601562e-06, "psnr_range_1_db": 135.28279036460714 } ] } ], "status": "exported" }, { "model": "rivagan_decoder", "legacy_file": "rivagan_decoder.onnx", "legacy_sha256": "6b006690cf352fa50a91a0b6d6241dbcded0a9957969e2c0665cf9a7011b3880", "variants": { "fp32": { "path": "rivagan_decoder_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 1088800, "sha256": "46354af2334b3339367c5fbe91d4261fdcddb63d5527bd4b2078222230f67375", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ] }, "fp16": { "path": "rivagan_decoder_fp16.onnx", "converter_fallback_reason": "Converter changed an external FP32 I/O type", "compute": "fp16_conv_matmul_gemm_fp32_other_ops", "fp16_compute_nodes": 4, "comparisons_to_fp32": [ [ { "shape": [ 1, 32 ], "mae": 0.011192433536052704, "max_abs": 0.049346923828125, "psnr_range_1_db": 35.46706765515768 } ], [ { "shape": [ 1, 32 ], "mae": 0.010502520715817809, "max_abs": 0.04539299011230469, "psnr_range_1_db": 35.95590597680599 } ], [ { "shape": [ 2, 32 ], "mae": 0.009412113577127457, "max_abs": 0.049346923828125, "psnr_range_1_db": 36.961599074282404 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 547053, "sha256": "e9a27f73d46fa67ac475b66f2c335d2b0fd528cc129bc7a188f5e49d2fe496b9", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ], "current_comparisons_to_fp32": [ [ { "shape": [ 1, 32 ], "mae": 0.011192433536052704, "max_abs": 0.049346923828125, "psnr_range_1_db": 35.46706765515768 } ], [ { "shape": [ 1, 32 ], "mae": 0.009129447862505913, "max_abs": 0.038016319274902344, "psnr_range_1_db": 38.296703601428995 } ], [ { "shape": [ 2, 32 ], "mae": 0.009412113577127457, "max_abs": 0.049346923828125, "psnr_range_1_db": 36.961599074282404 } ] ] }, "bf16": { "path": "rivagan_decoder_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 4, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 547156, "sha256": "40e1c0e2ce81eb3cd9380018f611a1984327c6362f980306e4ebc1c041213e5c", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 876444, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 32 ], "mae": 0.14347243681550026, "max_abs": 0.4973335266113281, "psnr_range_1_db": 14.464158887639924 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 32 ], "mae": 0.10220980240774225, "max_abs": 0.3697357177734375, "psnr_range_1_db": 17.032017323341943 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 32 ], "mae": 0.13584833405911922, "max_abs": 0.4973335266113281, "psnr_range_1_db": 15.055152820091177 } ] } ], "seconds": 2.18 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "legacy_fp32_onnx_graph", "sha256": "6b006690cf352fa50a91a0b6d6241dbcded0a9957969e2c0665cf9a7011b3880", "onnx_filename": "rivagan_decoder.onnx" }, "notes": [ "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 32 ] ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ [ 1, 32 ] ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 32 ] ] } ], "status": "exported", "validation_evidence": "Identical SHA-256 artifacts retain archived runtime/GPU checks; current CPU and source comparison rechecked.", "onnx_conversion_authorized_by_user": true, "original_onnx_reference_check": { "status": "passed", "source_sha256": "6b006690cf352fa50a91a0b6d6241dbcded0a9957969e2c0665cf9a7011b3880", "converted_sha256": "46354af2334b3339367c5fbe91d4261fdcddb63d5527bd4b2078222230f67375", "precision": "fp32", "input": [ 1, 3, 64, 64 ], "comparison": [ { "shape": [ 1, 32 ], "mae": 0.0, "max_abs": 0.0, "psnr_range_1_db": null } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "model": "rivagan_encoder", "legacy_file": "rivagan_encoder.onnx", "legacy_sha256": "66bc7814fadac7d686105c8ab26c6cf2cc8e40deb83e67bb6ef5b9de9b8ece0d", "variants": { "fp32": { "path": "rivagan_encoder_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 657157, "sha256": "04a3518a01f4d0af203b8e76d8e948dcbffcce777588bc0940689c2819bf583d", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] }, { "name": "data", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "output_height", "output_width" ] } ] }, "fp16": { "path": "rivagan_encoder_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 1, 64, 64 ], "mae": 9.584111732389525e-06, "max_abs": 0.00042006373405456543, "psnr_range_1_db": 93.43554503157749 } ], [ { "shape": [ 1, 3, 1, 64, 96 ], "mae": 9.859331511405697e-06, "max_abs": 0.0009808540344238281, "psnr_range_1_db": 91.56472881391448 } ], [ { "shape": [ 2, 3, 1, 64, 64 ], "mae": 1.0067886648812419e-05, "max_abs": 0.0006025433540344238, "psnr_range_1_db": 92.39948031481836 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 333475, "sha256": "e2a08357a4509c966c1d6f755cdd57345b22b7d8eccb9bbba6286404d4b79d12", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] }, { "name": "data", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "output_height", "output_width" ] } ], "current_comparisons_to_fp32": [ [ { "shape": [ 1, 3, 1, 64, 64 ], "mae": 9.584111732389525e-06, "max_abs": 0.00042006373405456543, "psnr_range_1_db": 93.43554503157749 } ], [ { "shape": [ 1, 3, 1, 80, 96 ], "mae": 9.538891734134975e-06, "max_abs": 0.0005042850971221924, "psnr_range_1_db": 93.42537951700885 } ], [ { "shape": [ 2, 3, 1, 64, 64 ], "mae": 1.0067886648812419e-05, "max_abs": 0.0006025433540344238, "psnr_range_1_db": 92.39948031481836 } ] ] }, "bf16": { "path": "rivagan_encoder_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 4, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 332787, "sha256": "acbf39e40d9db07ca0582b3c4b28623881e766252173a7e0afd14d092e5bdc03", "inputs": [ { "name": "frame", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "height", "width" ] }, { "name": "data", "dtype": "FLOAT", "shape": [ "batch", 32 ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, 1, "output_height", "output_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 726924, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 3, 1, 64, 64 ], "mae": 3.697259903390204e-05, "max_abs": 0.005499482154846191, "psnr_range_1_db": 73.04429602056754 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 3, 1, 64, 96 ], "mae": 4.260640500319823e-05, "max_abs": 0.009293019771575928, "psnr_range_1_db": 71.21712971808826 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 3, 1, 64, 64 ], "mae": 4.1072237147924774e-05, "max_abs": 0.006708979606628418, "psnr_range_1_db": 72.65050828531436 } ] } ], "seconds": 2.62 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "legacy_fp32_onnx_graph", "sha256": "66bc7814fadac7d686105c8ab26c6cf2cc8e40deb83e67bb6ef5b9de9b8ece0d", "onnx_filename": "rivagan_encoder.onnx" }, "notes": [ "User authorized direct conversion of existing ONNX on 2026-10-18 because original checkpoints are unavailable." ], "fp32_runtime_checks": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ [ 1, 3, 1, 64, 64 ] ] }, { "input": [ 1, 3, 80, 96 ], "outputs": [ [ 1, 3, 1, 80, 96 ] ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 1, 64, 64 ] ] } ], "status": "exported", "validation_evidence": "Identical SHA-256 artifacts retain archived runtime/GPU checks; current CPU and source comparison rechecked.", "onnx_conversion_authorized_by_user": true, "original_onnx_reference_check": { "status": "passed", "source_sha256": "66bc7814fadac7d686105c8ab26c6cf2cc8e40deb83e67bb6ef5b9de9b8ece0d", "converted_sha256": "04a3518a01f4d0af203b8e76d8e948dcbffcce777588bc0940689c2819bf583d", "precision": "fp32", "input": [ 1, 3, 64, 64 ], "comparison": [ { "shape": [ 1, 3, 1, 64, 64 ], "mae": 0.0, "max_abs": 0.0, "psnr_range_1_db": null } ], "tolerance": { "rtol": 1e-05, "atol": 1e-06 } } }, { "model": "SCUNet-GAN", "legacy_file": "SCUNet-GAN.onnx", "legacy_sha256": "79ae6073c91c2d25d1f199137a67c8d0f0807df27219cdd7d890f3cc6d5b43e7", "variants": { "fp32": { "path": "SCUNet-GAN_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 74443378, "sha256": "c9cce5f8ae05a8c9b2f58efe2b27f2cf8ada0ae386e060eb694ed39b303bb3f8", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "SCUNet-GAN_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 0.00012847230633875975, "max_abs": 0.00036972761154174805, "psnr_range_1_db": 76.41802641420897 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 0.0001268272567863783, "max_abs": 0.000554502010345459, "psnr_range_1_db": 75.97817788926955 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 0.00012831635103793815, "max_abs": 0.00036972761154174805, "psnr_range_1_db": 76.42306530333742 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 39744312, "sha256": "b328f6e6432555c72d0854dcba60cf44f42a8f9c408fae87f25c507baf81ccea", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "SCUNet-GAN_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 229, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 40076008, "sha256": "a59e83cc13c443f1f56f446ea361452947ef2463e581d12ca17045a6fe49cab8", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "892c83f812c59173273b74f4f34a14ecaf57a2fdb68df056664589beb55c966e", "architecture": "SCUNet", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "SCUNet-GAN.pth", "source_url": "https://github.com/cszn/KAIR/releases/download/v1.0/scunet_color_real_gan.pth" }, "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": 3.033055691048503e-07, "max_abs": 1.4901161193847656e-06, "psnr_range_1_db": 128.35931577555644 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 2.74122107334228e-07, "max_abs": 1.6689300537109375e-06, "psnr_range_1_db": 129.02511745976653 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 3.071957811092337e-07, "max_abs": 1.5497207641601562e-06, "psnr_range_1_db": 128.27006680857914 } ] } ], "status": "exported" }, { "model": "SCUNet-PSNR", "legacy_file": "SCUNet-PSNR.onnx", "legacy_sha256": "b0f8c12f1575bb49e39a85924152f1c6d4b527a4aae0432c9e5c7397123465e3", "variants": { "fp32": { "path": "SCUNet-PSNR_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 74443378, "sha256": "e68bc9c0dfa35e9a373a5a71c34d7cc8dbd3e56a48fbb702149b309438611b9b", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "fp16": { "path": "SCUNet-PSNR_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 3, 64, 64 ], "mae": 9.540992323309183e-05, "max_abs": 0.000362396240234375, "psnr_range_1_db": 78.55607028541768 } ], [ { "shape": [ 1, 3, 64, 96 ], "mae": 0.00012624326094131297, "max_abs": 0.0005705952644348145, "psnr_range_1_db": 75.99956857059783 } ], [ { "shape": [ 2, 3, 64, 64 ], "mae": 9.702681927592494e-05, "max_abs": 0.0003846883773803711, "psnr_range_1_db": 78.37337078853294 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 39744312, "sha256": "89f342963aece7a062b08e53d494369a0b96cbafcf22cb6653cb09df8cbe435d", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] }, "bf16": { "path": "SCUNet-PSNR_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 229, "runtime": "requires_native_gpu_validation", "status": "generated", "bytes": 40076008, "sha256": "6eacc01ecc5505bbcba3bad232dde07ffe8f0330e36ca37cecd3b380ac9470a6", "inputs": [ { "name": "input", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "output", "dtype": "FLOAT", "shape": [ "batch", 3, "output_height", "output_width" ] } ] } }, "low_bit_quantization": "requires_representative_calibration_images_and_quality_acceptance", "source": { "kind": "original_checkpoint", "sha256": "fa78899ba2caec9d235a900e91d96c689da71c42029230c2028b00f09f809c2e", "architecture": "SCUNet", "stored_floating_dtypes": [ "torch.float32" ], "checkpoint_filename": "SCUNet-PSNR.pth", "source_url": "https://github.com/cszn/KAIR/releases/download/v1.0/scunet_color_real_psnr.pth" }, "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": 2.1739348691577712e-07, "max_abs": 1.341104507446289e-06, "psnr_range_1_db": 131.07868183948065 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 3, 64, 96 ] ], "pytorch_comparison": [ { "shape": [ 1, 3, 64, 96 ], "mae": 2.4993642000481486e-07, "max_abs": 1.4901161193847656e-06, "psnr_range_1_db": 129.84544359870273 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 3, 64, 64 ] ], "pytorch_comparison": [ { "shape": [ 2, 3, 64, 64 ], "mae": 2.2153957009625932e-07, "max_abs": 1.3113021850585938e-06, "psnr_range_1_db": 130.965496166583 } ] } ], "status": "exported" }, { "model": "yolov5l-face", "legacy_file": "yolov5l-face.onnx", "legacy_sha256": "59ee8421e3d27a56f44d147e7ab6e4af07001de81bef2cd573c53d0fa03810dd", "variants": { "fp32": { "path": "yolov5l-face_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 186452424, "sha256": "792fbfe57cb04b299edca567ce0d512d22be492b2a80872d111a8995ba71e667", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ] }, "fp16": { "path": "yolov5l-face_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 48, 8, 8 ], "mae": 0.00032736193073408987, "max_abs": 0.004044055938720703, "psnr_range_1_db": 65.0351229149785 }, { "shape": [ 1, 48, 4, 4 ], "mae": 0.00020899780793115497, "max_abs": 0.0022096633911132812, "psnr_range_1_db": 68.00848549831855 }, { "shape": [ 1, 48, 2, 2 ], "mae": 8.003914263099432e-05, "max_abs": 0.0015516281127929688, "psnr_range_1_db": 71.80696751511245 } ], [ { "shape": [ 1, 48, 8, 12 ], "mae": 0.00042678983461680927, "max_abs": 0.006422996520996094, "psnr_range_1_db": 61.33877808524842 }, { "shape": [ 1, 48, 4, 6 ], "mae": 0.0003046116067303552, "max_abs": 0.005742073059082031, "psnr_range_1_db": 63.61167389588416 }, { "shape": [ 1, 48, 2, 3 ], "mae": 0.00021160084159216948, "max_abs": 0.0039234161376953125, "psnr_range_1_db": 63.47062586878482 } ], [ { "shape": [ 2, 48, 8, 8 ], "mae": 0.00042509638296905905, "max_abs": 0.009191274642944336, "psnr_range_1_db": 61.903312939317416 }, { "shape": [ 2, 48, 4, 4 ], "mae": 0.0002369930831870685, "max_abs": 0.002716064453125, "psnr_range_1_db": 67.11451453822752 }, { "shape": [ 2, 48, 2, 2 ], "mae": 9.948796165796618e-05, "max_abs": 0.002251148223876953, "psnr_range_1_db": 69.59587640635172 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 93293339, "sha256": "7979f49061d962510c04226e97e67218b43c0701ff44ecd467808d679d164e64", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ] }, "bf16": { "path": "yolov5l-face_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 112, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 93277858, "sha256": "98e0d8101ee2511cc2b19e93a120d21c7a1ae6191b9097dda6668dc11eca4e8e", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 94596444, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 48, 8, 8 ], "mae": 0.007373320087936008, "max_abs": 0.10570907592773438, "psnr_range_1_db": 37.569368328740005 }, { "shape": [ 1, 48, 4, 4 ], "mae": 0.003967481354872386, "max_abs": 0.04488372802734375, "psnr_range_1_db": 43.03384839972152 }, { "shape": [ 1, 48, 2, 2 ], "mae": 0.002335299572829778, "max_abs": 0.04443359375, "psnr_range_1_db": 43.21939307189527 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 48, 8, 12 ], "mae": 0.007517525182467782, "max_abs": 0.1102437973022461, "psnr_range_1_db": 36.43866594787514 }, { "shape": [ 1, 48, 4, 6 ], "mae": 0.004472517330820362, "max_abs": 0.07612419128417969, "psnr_range_1_db": 41.62557181181959 }, { "shape": [ 1, 48, 2, 3 ], "mae": 0.00261415667935378, "max_abs": 0.052257537841796875, "psnr_range_1_db": 43.37289275143903 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 48, 8, 8 ], "mae": 0.006904508401930798, "max_abs": 0.10570907592773438, "psnr_range_1_db": 38.00189065043083 }, { "shape": [ 2, 48, 4, 4 ], "mae": 0.003662808759448429, "max_abs": 0.05234050750732422, "psnr_range_1_db": 43.58219419434239 }, { "shape": [ 2, 48, 2, 2 ], "mae": 0.0023714266232370087, "max_abs": 0.054172515869140625, "psnr_range_1_db": 43.316209554358196 } ] } ], "seconds": 63.04 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "32787d8b8b936a98b1d166d923ec2892d5c962a1b3ab24582dda98cdcd3d6eae", "architecture": "YOLO5Face", "stored_floating_dtypes": [ "torch.float32" ], "upstream_checkpoint": { "model": "yolov5l-face", "sha256": "32787d8b8b936a98b1d166d923ec2892d5c962a1b3ab24582dda98cdcd3d6eae", "original_sha256": "adfa3fbee5ba97ca86237cf8b45992aaea891ea481d59722da89bbd871a6a546", "original_format": "upstream_pytorch_pt", "source_url": "https://github.com/clibdev/yolov5-face/releases/download/1.0.0/yolov5l-face.pt", "extraction": "Original FP32 state_dict and architecture YAML, without dtype conversion", "batchnorm_eps_for_app": 1e-05, "legacy_parameter_comparison": { "matched": 224, "total": 224 }, "network_commit": "152c688d551aefb973b7b589fb0691c93dab3564", "original_filename": "yolov5l-face-original.pth" }, "checkpoint_filename": "yolov5l-face.pth" }, "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, 48, 8, 8 ], [ 1, 48, 4, 4 ], [ 1, 48, 2, 2 ] ], "pytorch_comparison": [ { "shape": [ 1, 48, 8, 8 ], "mae": 4.0151038168308634e-07, "max_abs": 4.76837158203125e-06, "psnr_range_1_db": 123.87377816446008 }, { "shape": [ 1, 48, 4, 4 ], "mae": 2.423184923827648e-07, "max_abs": 3.814697265625e-06, "psnr_range_1_db": 126.4692411912121 }, { "shape": [ 1, 48, 2, 2 ], "mae": 8.296531935532887e-08, "max_abs": 1.9073486328125e-06, "psnr_range_1_db": 131.8229624324108 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 48, 8, 12 ], [ 1, 48, 4, 6 ], [ 1, 48, 2, 3 ] ], "pytorch_comparison": [ { "shape": [ 1, 48, 8, 12 ], "mae": 6.122092802090467e-07, "max_abs": 9.5367431640625e-06, "psnr_range_1_db": 119.42628688630292 }, { "shape": [ 1, 48, 4, 6 ], "mae": 2.9682996682822704e-07, "max_abs": 2.86102294921875e-06, "psnr_range_1_db": 125.21962219026277 }, { "shape": [ 1, 48, 2, 3 ], "mae": 1.136666267282433e-07, "max_abs": 1.9073486328125e-06, "psnr_range_1_db": 130.13596443984923 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 48, 8, 8 ], [ 2, 48, 4, 4 ], [ 2, 48, 2, 2 ] ], "pytorch_comparison": [ { "shape": [ 2, 48, 8, 8 ], "mae": 6.843392839073204e-07, "max_abs": 1.1444091796875e-05, "psnr_range_1_db": 118.72190253824458 }, { "shape": [ 2, 48, 4, 4 ], "mae": 4.6150792816964287e-07, "max_abs": 5.7220458984375e-06, "psnr_range_1_db": 121.47017712024888 }, { "shape": [ 2, 48, 2, 2 ], "mae": 2.492733983672224e-07, "max_abs": 3.814697265625e-06, "psnr_range_1_db": 124.14723343173779 } ] } ], "status": "exported" }, { "model": "yolov5n-face", "legacy_file": "yolov5n-face.onnx", "legacy_sha256": "edc42dea5f1bb4940a4ac5eb4374da5310db07c6b2ca60d2a065d3abe7c836f3", "variants": { "fp32": { "path": "yolov5n-face_fp32.onnx", "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 7015993, "sha256": "c9cc766b7ae5de148301398d65fb4b738790a98fb8347e2d1c4c0665fce1fc04", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ] }, "fp16": { "path": "yolov5n-face_fp16.onnx", "compute": "mixed_fp16_fp32_sensitive_ops", "comparisons_to_fp32": [ [ { "shape": [ 1, 48, 8, 8 ], "mae": 0.00045851000322727486, "max_abs": 0.0054264068603515625, "psnr_range_1_db": 62.60260261915403 }, { "shape": [ 1, 48, 4, 4 ], "mae": 0.00026286160088299465, "max_abs": 0.004776954650878906, "psnr_range_1_db": 64.82703260547889 }, { "shape": [ 1, 48, 2, 2 ], "mae": 0.00010178877225068088, "max_abs": 0.002010822296142578, "psnr_range_1_db": 70.6666764789767 } ], [ { "shape": [ 1, 48, 8, 12 ], "mae": 0.0005071538681578306, "max_abs": 0.007030487060546875, "psnr_range_1_db": 61.459071865030346 }, { "shape": [ 1, 48, 4, 6 ], "mae": 0.000315131697182854, "max_abs": 0.006817817687988281, "psnr_range_1_db": 64.09942458668047 }, { "shape": [ 1, 48, 2, 3 ], "mae": 0.00014279147015056678, "max_abs": 0.0025892257690429688, "psnr_range_1_db": 68.67711396724813 } ], [ { "shape": [ 2, 48, 8, 8 ], "mae": 0.00046997090491155785, "max_abs": 0.007456421852111816, "psnr_range_1_db": 62.146109536993094 }, { "shape": [ 2, 48, 4, 4 ], "mae": 0.0002569259280183663, "max_abs": 0.004776954650878906, "psnr_range_1_db": 65.6771312477511 }, { "shape": [ 2, 48, 2, 2 ], "mae": 9.038918142323382e-05, "max_abs": 0.002010822296142578, "psnr_range_1_db": 71.96744695768012 } ] ], "runtime": "passed_onnxruntime_cpu", "status": "generated", "bytes": 3649128, "sha256": "3eded7b7a0bc006063467a2756d08f03038057ccd277e6271c096ad8d7453eeb", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ] }, "bf16": { "path": "yolov5n-face_bf16.onnx", "compute": "bf16_conv_matmul_gemm_fp32_other_ops", "bf16_compute_nodes": 85, "runtime": "passed_tensorrt_gpu", "status": "generated", "bytes": 3597785, "sha256": "faa4e19e26c4c2d5044b21baa5bcba266d58abde926b4a68132f6410808b18d0", "inputs": [ { "name": "images", "dtype": "FLOAT", "shape": [ "batch", 3, "height", "width" ] } ], "outputs": [ { "name": "p3", "dtype": "FLOAT", "shape": [ "batch", 48, "p3_height", "p3_width" ] }, { "name": "p4", "dtype": "FLOAT", "shape": [ "batch", 48, "p4_height", "p4_width" ] }, { "name": "p5", "dtype": "FLOAT", "shape": [ "batch", 48, "p5_height", "p5_width" ] } ], "native_validation": { "tensorrt": "10.16.0.72", "gpu": "NVIDIA GeForce RTX 5070 Ti Laptop GPU", "builder_optimization_level": 0, "workspace_limit_bytes": 268435456, "tf32": false, "engine_bytes": 4983596, "status": "passed", "comparisons_to_fp32": [ { "input": [ 1, 3, 64, 64 ], "outputs": [ { "shape": [ 1, 48, 8, 8 ], "mae": 0.006569492497267977, "max_abs": 0.05897235870361328, "psnr_range_1_db": 40.268694369007804 }, { "shape": [ 1, 48, 4, 4 ], "mae": 0.0032434826134704053, "max_abs": 0.0639047622680664, "psnr_range_1_db": 44.5195918916059 }, { "shape": [ 1, 48, 2, 2 ], "mae": 0.002413822044521415, "max_abs": 0.03852510452270508, "psnr_range_1_db": 44.26082279129447 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ { "shape": [ 1, 48, 8, 12 ], "mae": 0.006645135144935921, "max_abs": 0.08284759521484375, "psnr_range_1_db": 39.92146797026477 }, { "shape": [ 1, 48, 4, 6 ], "mae": 0.005329395340393401, "max_abs": 0.07371234893798828, "psnr_range_1_db": 39.97066671131887 }, { "shape": [ 1, 48, 2, 3 ], "mae": 0.003591732199614247, "max_abs": 0.0625457763671875, "psnr_range_1_db": 40.3308924586114 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ { "shape": [ 2, 48, 8, 8 ], "mae": 0.006721253938546094, "max_abs": 0.09307193756103516, "psnr_range_1_db": 40.04561342555435 }, { "shape": [ 2, 48, 4, 4 ], "mae": 0.003893874447385315, "max_abs": 0.0639047622680664, "psnr_range_1_db": 43.23881753635297 }, { "shape": [ 2, 48, 2, 2 ], "mae": 0.0026677335639154385, "max_abs": 0.047100067138671875, "psnr_range_1_db": 43.43728172503103 } ] } ], "seconds": 32.29 } } }, "low_bit_quantization": "not_requested_for_light_models", "source": { "kind": "original_checkpoint", "sha256": "7920baabeedac685d43f716328047c077f76a766fae1b59a00d4f1d132a2f641", "architecture": "YOLO5Face", "stored_floating_dtypes": [ "torch.float32" ], "upstream_checkpoint": { "model": "yolov5n-face", "sha256": "7920baabeedac685d43f716328047c077f76a766fae1b59a00d4f1d132a2f641", "original_sha256": "794c94da54630f2ca66167fea25530c68133c61a2b14131b073c0d4064934e50", "original_format": "upstream_pytorch_pt", "source_url": "https://drive.usercontent.google.com/download?id=18oenL6tjFkdR1f5IgpYeQfDFqU4w3jEr&export=download&confirm=t", "extraction": "Original FP32 state_dict and architecture YAML, without dtype conversion", "batchnorm_eps_for_app": 1e-05, "legacy_parameter_comparison": { "matched": 170, "total": 170 }, "network_commit": "152c688d551aefb973b7b589fb0691c93dab3564", "original_filename": "yolov5n-face-original.pth" }, "checkpoint_filename": "yolov5n-face.pth" }, "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, 48, 8, 8 ], [ 1, 48, 4, 4 ], [ 1, 48, 2, 2 ] ], "pytorch_comparison": [ { "shape": [ 1, 48, 8, 8 ], "mae": 3.7755671655759215e-07, "max_abs": 3.814697265625e-06, "psnr_range_1_db": 124.55450468735498 }, { "shape": [ 1, 48, 4, 4 ], "mae": 2.1014905845125517e-07, "max_abs": 2.86102294921875e-06, "psnr_range_1_db": 127.8456396157592 }, { "shape": [ 1, 48, 2, 2 ], "mae": 1.810840331017971e-07, "max_abs": 3.337860107421875e-06, "psnr_range_1_db": 124.06639536802346 } ] }, { "input": [ 1, 3, 64, 96 ], "outputs": [ [ 1, 48, 8, 12 ], [ 1, 48, 4, 6 ], [ 1, 48, 2, 3 ] ], "pytorch_comparison": [ { "shape": [ 1, 48, 8, 12 ], "mae": 4.32026960576574e-07, "max_abs": 5.7220458984375e-06, "psnr_range_1_db": 122.3689933474968 }, { "shape": [ 1, 48, 4, 6 ], "mae": 2.720561395916674e-07, "max_abs": 5.7220458984375e-06, "psnr_range_1_db": 124.53477530189141 }, { "shape": [ 1, 48, 2, 3 ], "mae": 2.0595810686548552e-07, "max_abs": 3.814697265625e-06, "psnr_range_1_db": 124.3500898420559 } ] }, { "input": [ 2, 3, 64, 64 ], "outputs": [ [ 2, 48, 8, 8 ], [ 2, 48, 4, 4 ], [ 2, 48, 2, 2 ] ], "pytorch_comparison": [ { "shape": [ 2, 48, 8, 8 ], "mae": 6.384586725971531e-07, "max_abs": 9.5367431640625e-06, "psnr_range_1_db": 119.50844079042321 }, { "shape": [ 2, 48, 4, 4 ], "mae": 3.882284242232951e-07, "max_abs": 6.198883056640625e-06, "psnr_range_1_db": 122.78913928366605 }, { "shape": [ 2, 48, 2, 2 ], "mae": 2.1637233516003107e-07, "max_abs": 2.86102294921875e-06, "psnr_range_1_db": 126.11381851558912 } ] } ], "status": "exported" } ], "pending_models": [], "additional_dynamic_checks": { "BSRGANx2": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 160, 192 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 160, 192 ], "mae": 0.0007659037212660123, "max_abs": 0.0011948347091674805, "psnr_range_1_db": 62.25388845542816 } ] } }, "status": "passed" }, "BSRGANx4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00030988192296490526, "max_abs": 0.0015082526952028275, "psnr_range_1_db": 68.72171233830701 } ] } }, "status": "passed" }, "codeformer": { "status": "fixed_512_architecture" }, "codeformer_colorization": { "status": "fixed_512_architecture" }, "IRCNN_Lx1": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 80, 96 ], "mae": 6.28939577735663e-05, "max_abs": 0.00032973289489746094, "psnr_range_1_db": 82.0479804551795 } ] } }, "status": "passed" }, "IRCNN_Mx1": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 80, 96 ], "mae": 4.814838298443647e-05, "max_abs": 0.00023800134658813477, "psnr_range_1_db": 84.31794206695562 } ] } }, "status": "passed" }, "NAFNet-GoPro-width32": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 80, 96 ], "mae": 0.0015943397679721884, "max_abs": 0.01532745361328125, "psnr_range_1_db": 53.66691257847508 } ] } }, "status": "passed" }, "Real-CUGAN_v3_2x": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 160, 192 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 160, 192 ], "mae": 0.04583884990877575, "max_abs": 0.157562255859375, "psnr_range_1_db": 25.09526840748551 } ] } }, "status": "passed" }, "Real-CUGAN_v3_3x": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 240, 288 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 240, 288 ], "mae": 0.08290176181881516, "max_abs": 0.576171875, "psnr_range_1_db": 20.072286427649317 } ] } }, "status": "passed" }, "Real-CUGAN_v3_4x": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.08824567772216041, "max_abs": 1.104095458984375, "psnr_range_1_db": 18.846541257738835 } ] } }, "status": "passed" }, "Real_ESRGAN_x4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00018638396116633278, "max_abs": 0.0010055005550384521, "psnr_range_1_db": 73.10602769024183 } ] } }, "status": "passed" }, "Real_ESRGAN_x4plus": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00018638396116633278, "max_abs": 0.0010055005550384521, "psnr_range_1_db": 73.10602769024183 } ] } }, "status": "passed" }, "RealESR_Animex4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.0002476347888609275, "max_abs": 0.0018429160118103027, "psnr_range_1_db": 70.01847662978355 } ] } }, "status": "passed" }, "RealESR_Animex4_v2": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00032935698492085266, "max_abs": 0.002398192882537842, "psnr_range_1_db": 67.50808858600125 } ] } }, "status": "passed" }, "RealESR_Gx4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.000277196236821611, "max_abs": 0.0029249340295791626, "psnr_range_1_db": 68.73033814061857 } ] } }, "status": "passed" }, "RealESR_Gx4_WDN": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.0006655417581959631, "max_abs": 0.011030018329620361, "psnr_range_1_db": 60.741862121473595 } ] } }, "status": "passed" }, "Restormer-Defocus": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 80, 96 ], "mae": 9.29726539955785e-05, "max_abs": 0.0005126595497131348, "psnr_range_1_db": 78.59869848709211 } ] } }, "status": "passed" }, "Restormer-Motion": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 80, 96 ], "mae": 0.00012211790275008146, "max_abs": 0.0007197856903076172, "psnr_range_1_db": 76.1495878089218 } ] } }, "status": "passed" }, "SCUNet-GAN": { "input": [ 1, 3, 128, 64 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 128, 64 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 128, 64 ], "mae": 0.00012667574507455961, "max_abs": 0.00035959482192993164, "psnr_range_1_db": 76.55023841254857 } ] } }, "status": "passed" }, "SCUNet-PSNR": { "input": [ 1, 3, 128, 64 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 128, 64 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 128, 64 ], "mae": 8.971942103623103e-05, "max_abs": 0.00041049718856811523, "psnr_range_1_db": 79.00186802285171 } ] } }, "status": "passed" }, "yolov5l-face": { "input": [ 1, 3, 96, 128 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 48, 12, 16 ], [ 1, 48, 6, 8 ], [ 1, 48, 3, 4 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 48, 12, 16 ], "mae": 0.0002831699069550571, "max_abs": 0.0067321062088012695, "psnr_range_1_db": 64.86164202891834 }, { "shape": [ 1, 48, 6, 8 ], "mae": 0.000254871191120603, "max_abs": 0.004535675048828125, "psnr_range_1_db": 66.35413702225931 }, { "shape": [ 1, 48, 3, 4 ], "mae": 0.00017773665604181588, "max_abs": 0.0036139488220214844, "psnr_range_1_db": 64.66252819511276 } ] } }, "status": "passed" }, "yolov5n-face": { "input": [ 1, 3, 96, 128 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 48, 12, 16 ], [ 1, 48, 6, 8 ], [ 1, 48, 3, 4 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 48, 12, 16 ], "mae": 0.000545447458635964, "max_abs": 0.00852346420288086, "psnr_range_1_db": 60.40003290990592 }, { "shape": [ 1, 48, 6, 8 ], "mae": 0.0004008672306857382, "max_abs": 0.0069255828857421875, "psnr_range_1_db": 62.93851516054434 }, { "shape": [ 1, 48, 3, 4 ], "mae": 0.00022013775419004055, "max_abs": 0.00327301025390625, "psnr_range_1_db": 66.2009864819302 } ] } }, "status": "passed" }, "DeblurGANv2-FPNInception-GoPro": { "input": [ 1, 3, 320, 256 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 256 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 256 ], "mae": 3.6088456685826977e-05, "max_abs": 0.0009170174598693848, "psnr_range_1_db": 81.9120505985635 } ] } }, "status": "passed" }, "Deblurgan_v2": { "input": [ 1, 3, 320, 256 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 256 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 256 ], "mae": 0.00011104988066001908, "max_abs": 0.0006253719329833984, "psnr_range_1_db": 76.91790856735126 } ] } }, "status": "passed" }, "face_enhance_x4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 0.00011088519600485192, "max_abs": 0.00078544020652771, "psnr_range_1_db": 77.17037861271258 } ] } }, "status": "passed" }, "LVAx2": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 160, 192 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 160, 192 ], "mae": 7.866326272455303e-05, "max_abs": 0.000244140625, "psnr_range_1_db": 79.99381254025931 } ] } }, "status": "passed" }, "MSharpx4": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 320, 384 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 320, 384 ], "mae": 9.23772332902849e-05, "max_abs": 0.0005388855934143066, "psnr_range_1_db": 78.61247667424611 } ] } }, "status": "passed" }, "rivagan_decoder": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 32 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 32 ], "mae": 0.009129447862505913, "max_abs": 0.038016319274902344, "psnr_range_1_db": 38.296703601428995 } ] } }, "status": "passed" }, "rivagan_encoder": { "input": [ 1, 3, 80, 96 ], "variants": { "fp32": { "status": "passed", "outputs": [ [ 1, 3, 1, 80, 96 ] ] }, "fp16": { "status": "passed", "comparison": [ { "shape": [ 1, 3, 1, 80, 96 ], "mae": 9.538891734134975e-06, "max_abs": 0.0005042850971221924, "psnr_range_1_db": 93.42537951700885 } ] } }, "status": "passed" } } }