ModelTC/MQBench

关于yolov5s进行PTQ量化出现TraceError问题

Closed this issue · 3 comments

xiaopengaia commented

嗨 大家好，

今天我尝试使用mqbench对yolov5s进行PTQ量化

yolov5s模型来自于：https://github.com/ultralytics/yolov5.git

当我尝试如下代码进行yolov5s量化处理时

from mqbench.prepare_by_platform import prepare_by_platform, BackendType

backend = BackendType.ONNX_QNN
model = prepare_by_platform(model, backend)

出现了这个问题

torch.fx.proxy.TraceError: symbolically traced variables cannot be used as inputs to control flow

请问大家，这个问题有什么好的，简便的方式处理呢？

aksenventwo commented

可以看看 #82

xiaopengaia commented

好的，非常感谢，这个对我很有帮助

github-actions commented

This issue has not received any updates in 120 days. Please reply to this issue if this still unresolved!