当前位置: 首页 > 工具软件 > MMdnn > 使用案例 >

mmdnn模型转换到caffe,卷积层convolution_param中添加engine: CAFFE

向嘉誉
2023-12-01

mmdnn模型转换到caffe,卷积层convolution_param中添加engine: CAFFE

在mmdnn->conversion->caffe的caffe——emitter.py 中修改
def emit_Conv(self,IR_node)
修改self.add_body(),在ntop=1后面添加 engine=1

 类似资料: