From 3e0dfec4093ab3bd3192b7c3253c83176913130a Mon Sep 17 00:00:00 2001 From: icolbert Date: Wed, 13 Sep 2023 21:00:55 -0700 Subject: [PATCH] Fixing 14b -> 13b --- src/brevitas_examples/super_resolution/README.md | 2 +- src/brevitas_examples/super_resolution/models/__init__.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/brevitas_examples/super_resolution/README.md b/src/brevitas_examples/super_resolution/README.md index 55f1b5d75..9e0068c27 100644 --- a/src/brevitas_examples/super_resolution/README.md +++ b/src/brevitas_examples/super_resolution/README.md @@ -23,7 +23,7 @@ Note that this is a difference from many academic works that train only on the Y || | [quant_espcn_x2_w4a4_base]() | x2 | int4 | (u)int4 | 30.30 | | [quant_espcn_x2_w4a4_a2q_32b]() | x2 | int4 | (u)int4 | 30.27 | -| [quant_espcn_x2_w4a4_a2q_14b]() | x2 | int4 | (u)int4 | 30.24 | +| [quant_espcn_x2_w4a4_a2q_13b]() | x2 | int4 | (u)int4 | 30.24 | ## Train diff --git a/src/brevitas_examples/super_resolution/models/__init__.py b/src/brevitas_examples/super_resolution/models/__init__.py index f3c9a0ad5..872624a85 100644 --- a/src/brevitas_examples/super_resolution/models/__init__.py +++ b/src/brevitas_examples/super_resolution/models/__init__.py @@ -37,13 +37,13 @@ weight_bit_width=4, act_bit_width=4, acc_bit_width=32), - 'quant_espcn_x2_w4a4_a2q_14b': + 'quant_espcn_x2_w4a4_a2q_13b': partial( quant_espcn_a2q, upscale_factor=2, weight_bit_width=4, act_bit_width=4, - acc_bit_width=14)} + acc_bit_width=13)} root_url = 'https://github.com/Xilinx/brevitas/releases/download/super_res-r0'