关于人工智能:OneFlowONNX-v060正式发布

37次阅读

共计 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…

正文完
 0