• MSAdapterModelZoo
• • • Table of Contents
  • Officical
  • • cv
  • Research
  • Tools

MSAdapterModelZoo

MSAdapterModelZoo 存放 MSAdapter 已适配的PyTorch网络结构。

MSAdapter是MindSpore适配PyTorch接口的工具,其目的是在不改变原有PyTorch用户的使用习惯情况下,使得PyTorch代码能在昇腾上获得高效性能.

Table of Contents

• Officical
• Research
• Tools

Officical

official/ 内放置MSAdapter适配的Torch-Code-Style可训练模型代码。

命名规则如下：

XX_torch/ 或 XX_torch.py是 算法的Torch官方代码或者Github上高星项目源码。

XX_adapter/ 或 XX_adapter.py 是适配后可以调用MindSpore框架在Ascend/GPU/CPU上train/eval的代码。

cv

official/cv/ 内放置MSAdapter适配的CV领域可训练算法代码。

目前支持的模型如下：

• alexnet
• darknet
• mobilenet_v2
• ssd

Research

research/ 内放置MSAdapter适配的Torch-Code-Style可推理模型代码。

Tools

tools/ 内放置在MSAdapter适配模型过程中可能遇到的工具。

tools/pth2ckpt 可以将Torch的模型权重文件.pth转换为MindSpore使用的模型权重文件格式.ckpt。

tools/convert_to_ms_tensor可以将Torch的Tensor格式转换为对应的MindSpore Tensor格式。

以上工具的使用方法在tools/的readme有详细教程。