共计 1237 个字符,预计需要花费 4 分钟才能阅读完成。
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…
正文完