找了2个多小时的代码报错问题终于解决,纪念一下

最终原因还是自己构建数据库不熟练,┭┮﹏┭┮

首先,报错信息:IndexError: index 4392 is out of bounds for axis 0 with size 20。

意思是数组第一个维度的指数溢出了,比如数组a第一维度的指数只有20,但是出现了 a【4392】,故报错。

在看到这行报错,我完全不知所措,不知道怎么下手,反复查看我的dataset中的data有没有构建错,完全没有想到是label出错了,因为是做自编码,是无监督学习,不需要label的参与,所以完全没有重视,写到这行的时候,没有仔细思考,要引以为戒,思路一定要清晰。

最后该怎么修改呢,

label = np.zeros([num, length])

应该改成

label = np.zeros([num, 1])

返回给数据库的data和target分别如下

# data[num,len],label[num]

你可能感兴趣的:(python,开发语言,pytorch)