Skip to content

mgeconvert v0.6.0

Compare
Choose a tag to compare
@daisycx daisycx released this 14 Dec 03:38
· 57 commits to master since this release

HighLight

  • 支持1.7版本的MegEngine
  • TracedModule QAT模型转caffe支持导出量化参数文件
  • 支持安装时使用 all 选项安装所有转换器

Bug Fixes

  • 修复convert脚本中的bug,统一TracedModule转换器接口
  • 修复各后端的安装脚本,解决依赖版本冲突问题,支持安装时使用 all 选项
  • 依赖的protobuf升级至3.11.1版本,使用USE_CAFFE_PROTO环境变量管理 proto版本
  • 修复tflite转换器中depthwise conv参数depth_multiplier和conv的pad mode计算错误
  • 使用onnxoptimizer替换onnx.optimizer,修复新版本onnx移除optimizer导致的不兼容问题

New Features

  • 兼容MegEngine 1.7的TracedModule
  • TracedModule QAT模型转caffe支持导出量化参数文件
  • caffe转换器支持指定转换后端,包括caffe、snpe、trt、nnie
  • convert命令支持自动补全
  • TracedModule模型支持指定end points转换部分网络
  • 添加新transform支持:融合Conv和BN、融合Linear/MatMul和BN、去除Linear/MatMul之前的Flatten算子
  • 支持新算子的转换:relu6、ConvBn、ConvRelu、ConvBnRelu
  • 量化数据类型支持MegEngine内置的类型之外的其他numpy.dtype
  • 添加api接口说明wiki