作为当前AI领域最流行的框架之一,PyTorch经过寒武纪的巧妙定制化,催生了Cambricon PyTorch,专门为寒武纪加速板卡和AI软件栈而设计。与原生PyTorch相比,用户无需改动任何代码,即可快速将AI模型迁移到Cambricon PyTorch平台。该仓库详细示范了在CV分类、检测、分割、NLP、语音等领域中常用的各类经典和前沿AI模型的适配方法,使其在Cambricon PyTorch上高效运行。对于开发者而言,参考本仓库可极大地简化其他AI应用的移植过程,实现更便捷、高效的部署。
Cambricon PyTorch:高效移植AI模型的开发利器
文件列表
PyTorch是时下最流行的AI框架,cambriocn pytorch训练和推理模型集合
(预估有个3079文件)
configure
1KB
.clang-format
1KB
lstm.cpp
12KB
taco2ProjectionLayerPlugin.cpp
18KB
taco2AttentionLayerPlugin.cpp
19KB
taco2PrenetLayerPlugin.cpp
13KB
taco2LSTMCellLayerPlugin.cpp
18KB
taco2ModulationRemovalLayerPlugin.cpp
12KB
decoderBuilderPlain.cpp
18KB
decoderBuilderPlugins.cpp
21KB
暂无评论