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