Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
|
1 day ago | |
---|---|---|
official | 1 day ago | |
research | 1 day ago | |
tools | 1 day ago | |
.gitignore | 3 months ago | |
README.md | 3 months ago |
MSAdapterModelZoo 存放 MSAdapter 已适配的PyTorch网络结构。
MSAdapter是MindSpore适配PyTorch接口的工具,其目的是在不改变原有PyTorch用户的使用习惯情况下,使得PyTorch代码能在昇腾上获得高效性能.
official/
内放置MSAdapter适配的Torch-Code-Style可训练模型代码。
命名规则如下:
XX_torch/
或 XX_torch.py
是 算法的Torch官方代码或者Github上高星项目源码。
XX_adapter/
或 XX_adapter.py
是适配后可以调用MindSpore框架在Ascend/GPU/CPU上train/eval的代码。
official/cv/
内放置MSAdapter适配的CV领域可训练算法代码。
目前支持的模型如下:
research/
内放置MSAdapter适配的Torch-Code-Style可推理模型代码。
tools/
内放置在MSAdapter适配模型过程中可能遇到的工具。
tools/pth2ckpt
可以将Torch的模型权重文件.pth
转换为MindSpore使用的模型权重文件格式.ckpt
。
tools/convert_to_ms_tensor
可以将Torch的Tensor格式转换为对应的MindSpore Tensor格式。
以上工具的使用方法在tools/
的readme有详细教程。
No Description
Jupyter Notebook Text C++ Python C other