主要是网络那部分的问题,我之前是二分类,在全连接层那块
self.out = nn.Linear(10, 2)
但是还数据了,分21类,需要将全连接层输入大于分类的个数,否则报错,于是修改为
self.out = nn.Linear(64, 21)
解决
主要是网络那部分的问题,我之前是二分类,在全连接层那块
self.out = nn.Linear(10, 2)
但是还数据了,分21类,需要将全连接层输入大于分类的个数,否则报错,于是修改为
self.out = nn.Linear(64, 21)
解决