New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

rv1126混合量化后维度异常 #478

Open

zj445680666 opened this issue Oct 11, 2024 · 1 comment

zj445680666 commented Oct 11, 2024

混合量化前输入时1x3x640x640.混合量化后，输入变为640x640x3x16了。

我只是混合量化在自定义处随意添加了一层进行float32不量化。

结果维度就异常了。但是使用官方那个pb模型又是可以的。使用yolov5和yolov8都是一样的现象。

BITfanyu commented Nov 15, 2024

我的是batch变成100了，在step1 rknn.load的时候加入input_size_list=参数就正常了

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment