亮点特性
继承开源PyTorch特性,适配昇腾AI处理器系列;实现使用昇腾AI处理器进行动态图训练,现有网络只修改设备类型和数据类型,即可迁移到昇腾AI处理器上使用;支持自定义算子开发和模型调整
适配昇腾AI处理器
基于开源PyTorch,适配昇腾AI处理器,提供原生Python接口
框架基础功能
PyTorch动态图、自动微分、Profiling、优化器等
自定义算子开发
支持在PyTorch框架中添加自定义算子
分布式训练
支持原生分布式数据并行训练,包含单机多卡、多机多卡场景支持的集合通信原语包含Broadcast、AllReduce等
混合精度
支持O1、O2混合精度和静态动态Loss Scale算法,并且能够通过黑白名单配置算子的精度。立即下载、安装Apex混合精度优化包
模型推理
支持输出标准的ONNX模型可通过离线转换工具将ONNX模型转换为离线推理模型