Releases · Oneflow-Inc/oneflow_convert

v0.6.0发布

v0.6.0版本相比于v0.5.4版本提升了转换接口的易用性，开发了多个新的特性。并新增了6种模型的支持以及20余种Op的支持，以及修复6个转换过程中的bug。版本发布过程中，感谢以下贡献者的支持。 @Flowingsun007 @doombeaker @liujuncheng @leaves-zwx @CPFLAME @BBuf @zhongshsh

pip install oneflow-onnx==0.6.0 一键安装使用 🎉🎉🎉 。

新增支持的模型

新增支持 mobilenetv3。#63
新增支持 squeezenet 。#63
新增支持 yolov5。#95
新增支持 disco diffusion。#94
新增支持 libai 仓库的 T5 模型。#92
新增支持 libai 仓库的 VisionTransformer 模型。#103

新增支持的Op

新增支持hard_swish和hard_sigmoid op。#63
新增支持arange op。#71
新增支持 expand_dims op。#74
新增支持 narrow op。#75
新增支持 silu op。#78
新增支持 upsample_nearest_2d op。#79
新增支持 var op。#81
新增支持 conv1d op。#83
新增支持 scalar_div op。#84
新增支持 cublas_fused_mlp op。#90
新增支持 elementwise max/min op。#92
新增支持 broadcast_matmul，where，scalar_logical_less，scalar_logical_greater，gather op。#93
新增 expand op。#95
新增支持 fill_ op。#96
新增支持gelu op。#100
新增支持layernorm op。#101
新增支持amp_white_identity/amp_black_identity op。#102

BUG 修复

修复安装oneflow_onnx时不自动安装onnx，onnxruntime等依赖包的问题。#60
修复因为版本更新导致的maxpool op转换失败的bug。#61
修复unsqueeze op在Opset 13下的bug。#65
修复获取tensor时硬编码导致错误。#77
修复保存超大tensor时的size推导错误。#82
修复了在libai下, 用T5做test 转onnx时遇到的问题。（支持了oneflow在编译graph时,采用global tensor进行推理的写法以及flow.bool类型Tensor作为输入）#88
修复 pool 多了一个index 输出导致TensorRT推理失败的问题。#97