用cnn进行人脸识别时的报错 - pytorch吧

level 1

CARROT♬ 楼主

outputs = net(inputs)
print(outputs.shape)
labels = Variable(labels.type(torch.LongTensor))
print(labels)
loss = criterion(outputs, labels)
loss.backward()
optimizer.step()
初学pytorch的萌新。以上为cnn的训练部分，运行时在loss = criterion(outputs, labels)这一行会报错，显示IndexError: Target xx is out of bounds.程序在图像少于三十张左右时可以正常运行，但再多就会出现这个错误。请问有人遇到过相同的问题吗

2020年02月18日 02点02分 1

level 1

居一包😺🐯

请问你的问题解决了嘛我也遇到了这个问题

2020年05月05日 05点05分 2

CARROT♬

解决了，就是全连接最后一层的类别数忘改了 [喷]

2020年05月05日 07点05分

居一包😺🐯

@CARROT♬ 请问你pytorch用的好吗可以提供有偿指导吗我实在整不出来了😂 。我在用pytorch搭建一个很简单的网络但是一到损失函数就报错，还想把数据最后进行可视化

2020年05月05日 07点05分

CARROT♬

@居一包😺🐯 其实我也刚学不久

2020年05月05日 10点05分

居一包😺🐯

@CARROT♬ 哦哦哎我这遇到问题就解决不掉很焦虑😂

2020年05月05日 10点05分

level 1

hsuihiuwubw

请问是类别数改成什么呢
困扰好久了，希望获得解答，谢谢！

2020年07月08日 13点07分 3

CARROT♬

图上代码没错，我错的是全连接层。我那个程序的训练集有20类，全连接层是从别的程序粘过来的，忘了改参数，最后一层只有10类，所以报错，后来改成20就正常了。

2020年07月08日 13点07分

hsuihiuwubw

@CARROT♬ 对，我想知道是把self.fc1 = nn.Linear(16 * 53 * 53, 1024)self.fc2 = nn.Linear(1024, 512)self.fc3 = nn.Linear(512,2) 最后的Linear(512,2)改成（512，3）吗？如果训练集有3类？

2020年07月08日 14点07分

hsuihiuwubw

@CARROT♬ 谢谢！我解决了

2020年07月08日 15点07分

level 1

hsuihiuwubw

能参考一下您的全连接层那部分代码的修改过程吗，就是原代码和修改后的代码，就那一句就行，我的现在能训练2类但是3类就不行了

2020年07月08日 14点07分 4