v0.6.0

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

新Feature

• 重构代码仓库的示例文档。#68
• 重构导出onnx api，flow_weight_dir参数可选，提升易用性。#73
• 允许mapping过程中访问到原始op_node以获得更多必要信息。#80
• CI支持black 格式化。#86
• 支持Graph里面有Free Eager Tensor。#89
• 转换onnx时支持多个输入的Graph对象。#99