You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
laich 6642ca67ff Fix format (#55) 1 day ago
ms_adapter Fix format (#55) 1 day ago
testing add register buffer, fix *arg bug (#54) 1 day ago
.gitignore workflow example ResNet training (#29) 3 weeks ago
LICENSE Add information 1 month ago
README.md Add information 1 month ago
SupportedList.md Add information 1 month ago
requirements.txt Add information 1 month ago
setup.py Add information 1 month ago

README.md

MSAdapter

简体中文 | English

MindSpore适配PyTorch接口工具。

简介

MSAdapter可以使PyTorch的模型在昇腾(Ascend)上运行。

  • 模型支持: MSAdapter的目的是支持PyTorch语法的原生态表达,用户只需要将PyTorch源代码中import torch替换为import ms_adapter.pytorch即可实现模型能支持昇腾上训练。
    模型中所使用的APIs支持状态可以从这里找到 Supported List.

安装

通过pip安装

pip install ms_adapter

通过源码安装

 git clone https://git.openi.org.cn/OpenI/MSAdapter.git
 cd MSAdapter
 python setup.py install

使用

引用

简介

MindSpore对PyTorch接口的支持工具

Python Markdown