MindSpore：Ascend芯片Mindspore代码实现小样本图像分类，在网络训练时出现错误

BaldheadedM

于 2022-08-03 21:35:28 发布

阅读量218

点赞数

文章标签：分类 python 深度学习

本文链接：https://blog.csdn.net/BaldheadedM/article/details/126149456

版权

在Mindspore v1.0.1和Ascend环境下，进行小样本图像分类时遇到错误。使用GeneratorDataset和自定义Sampler、WithLossCell、TrainOneStepCell后，网络训练时报错。错误信息涉及函数'function' object没有属性'name'。解决方案是升级到Mindspore 1.1及以上版本，该问题已在新版本中修复。

摘要由CSDN通过智能技术生成

问题描述：

Mindspore环境：v1.0.1，Ascend

问题背景：小样本图像分类，需要自定义数据集，采用了GeneratorDataset；自定义采样器Sampler；仿照教程自定义了WithLossCell和TrainOneStepCell，为了以循环的方式

进行网络的训练，发现报错

报错信息如下：

[ERROR] PIPELINE(13373,python):2021-04-10-08:30:55.177.577 [mindspore/ccsrc/pipeline/jit/pipeline.cc:536] Compile]
Traceback (most recent call last):
File "train.py", line 242, in <module>
main()
File "train.py", line 225, in main
optim=optim)
File "train.py", line 147, in train
loss, acc = net(inputs["img"], inputs["label"], opt.num_support_tr)
File "/usr/local/lib/python3.7/dist-packages/mindspore/nn/cell.py", line 280, in __call__
out = self.compile_and_run(*inputs)
File "/usr/local/lib/python3.7/dist-packages/mindspore/nn/cell.py", line 535, in compile_