OneFlow-ONNX v0.6.0正式公布。新版本晋升了转换接口的易用性,开发了多个新个性,并新增反对6种模型以及20多种算子,此外,还修复了6个转换过程中的bug。更新详情请查看链接:https://github.com/Oneflow-In...

欢送通过下列代码一键装置应用,期待你的反馈。
pip install oneflow-onnx==0.6.0

以下是本次更新次要内容。

新增反对的模型

  • 新增反对MobileNetv3
  • 新增反对SqueezeNet
  • 新增反对YOLOv5
  • 新增反对Disco Diffusion
  • 新增反对LiBai仓库的T5模型
  • 新增反对LiBai仓库的VisionTransformer 模型

新增反对的Op

  • 新增反对hard_swish和hard_sigmoid op
  • 新增反对arange op
  • 新增反对expand_dims op
  • 新增反对narrow op
  • 新增反对silu op
  • 新增反对upsample_nearest_2d op
  • 新增反对var op
  • 新增反对conv1d op
  • 新增反对scalar_div op
  • 新增反对cublas_fused_mlp op
  • 新增反对elementwise max/min op
  • 新增反对broadcast_matmul,where,scalar_logical_less,scalar_logical_greater,gather op
  • 新增expand op
  • 新增反对fill_ op
  • 新增反对gelu op
  • 新增反对layernorm op
  • 新增反对amp_white_identity/amp_black_identity op

BUG 修复

  • 修复装置oneflow_onnx时不主动装置onnx、onnxruntime等依赖包的问题
  • 修复因为版本更新导致的maxpool op转换失败的bug
  • 修复unsqueeze op在Opset 13下的bug
  • 修复获取tensor时硬编码导致谬误
  • 修复保留超大tensor时的size推导谬误
  • 修复了在LiBai下, 用T5做test转ONNX时遇到的问题。(反对了OneFlow在编译Graph时,采纳Global Tensor进行推理的写法以及flow.bool类型Tensor作为输出)
  • 修复 pool 多了一个index 输入导致TensorRT推理失败的问题

新增Feature

  • 重构代码仓库的示例文档
  • 重构导出onnx api,flow_weight_dir参数可选,晋升易用性
  • 容许mapping过程中拜访到原始op_node以取得更多必要信息
  • CI反对black格式化
  • 反对Graph外面有Free Eager Tensor
  • 转换ONNX时反对多个输出的Graph对象

版本公布过程中,感激以下贡献者的反对 :@Flowingsun007 、@doombeaker 、@liujuncheng、@leaves-zwx 、@CPFLAME、@BBuf、@zhongshsh

欢送下载体验 OneFlow v0.8.0 最新版本:https://github.com/Oneflow-In...