华人澳洲中文论坛

热图推荐

    OneFlow-ONNX v0.6.0正式公布

    [复制链接]

    2022-11-25 07:02:35 21 0



    OneFlow-ONNX v0.6.0正式公布。新版本晋升了转换接口的易用性,开发了多个新特性,并新增反对6种模型以及20多种算子,另外,还修复了6个转换过程当中的bug。更新详情请查看链接:http://github.com/Oneflow-Inc/oneflow_convert/releases/tag/v0.6.0
    欢送经过以下代码一键装置使用,期待你的反馈。
    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 opBUG 修复
    修复装置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 最新版本:http://github.com/Oneflow-Inc/oneflow/

    发表回复

    您需要登录后才可以回帖 登录 | 立即注册

    返回列表 本版积分规则

    :
    注册会员
    :
    论坛短信
    :
    未填写
    :
    未填写
    :
    未填写

    主题27

    帖子34

    积分160

    图文推荐