Dsp Tian

摘要： 如果从网上down下来的onnx文件的batchsize不是1或者是动态batchsize。 下面代码可以强行将batchsize设为1，方便推理。 代码如下： import onnx # 1. 加载原始模型 model = onnx.load("feature.onnx") # 2. 修改输入层b 阅读全文