Multi-input network model, quantization error occurred

Xilinx / Vitis-AI

Vitis AI is Xilinx’s development stack for AI inference on Xilinx hardware platforms, including both edge devices and Alveo cards.

https://www.xilinx.com/ai

Apache License 2.0

1.49k stars 630 forks source link

Multi-input network model, quantization error occurred #1438

Closed Ruiii1 closed 5 months ago

Ruiii1 commented 5 months ago

Hello everyone I encountered an error while quantizing my network. My network requires three input parameters (a, b, c) and outputs one parameter (y). When I use nndct_nn.QuantStub and nndct_nn.DeQuantStub to quantize (a), I am unable to use the other two parameters (b and c). How should I solve this issue? I would appreciate any guidance or suggestions. Thank you very much!