Altas 200I DK A2中的Ascend 310只支持推理,不支持NPU板端训练吗?

Altas 200I DK A2开发板的NPU型号为310系列,该系列NPU只能进行推理,不能进行板端训练吗?

用户您好,目前Ascend910系列可用于训练和推理,Ascend310只能用于推理。

其实是支持的,框架层是支持训练机制的,但由于310b反向算子缺失比较多,再加上显存内存共用,训练消耗内存很大,估计能跑起来的模型没几个,resnet50+cifar10数据集我跑过训练,mobilenet就不行,缺反向算子,大点的模型或者图片尺寸较大的数据集大概率内存不足,且能跑的通常也要开虚拟内存swap;

好的,感谢。

好的,感谢您的回答。

此话题已在最后回复的 60 分钟后被自动关闭。不再允许新回复。