OneFlow-ONNX v0.6.0 正式发布
OneFlow-ONNX v0.6.0 正式发布。新版本提升了转换接口的易用性，开发了多个新特性，并新增支持 6 种模型以及 20 多种算子，此外，还修复了 6 个转换过程中的 bug。更新详情请查看链接：https://github.com/Oneflow-Inc/oneflow_convert/releases/tag/v0.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-Inc/oneflow/
不至于成为世界上最快的深度学习框架。 2022-03-23 加入
★ OneFlow深度学习框架：github.com/Oneflow-Inc/oneflow ★ OF云平台：oneflow.cloud
评论