PyTorch | PyTorch实战过程中遇到的问题记录与总结

本篇的普适度不高，都是我个人在一个特定项目中遇到的问题的记录，以防再次出现相同错误时，还要花费力气去查找。

你可以试着用ctrl+F，输入你想要搜索的关键词，快速确定这里有没有你想要的信息。Have a nice day ^-^

1. RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

本来打算在AUC>threshold的时候在去算AUC2的值，

于是有了下面的代码：

for idx,val in enumerate(self.gt2_pre_labels):
	self.gt2_pre_labels = self.an_pre_labels.mul(torch.argmax(self.gt2_pre_labels_tmp,dim=1))
      if self.gt2_labels[idx] == self.gt2_pre_labels[idx]:

sum += 1
auc2 = float(sum) / float(len(self.dataloader['test'].dataset))

运行就出现了上面的现象，而且前几个epoch还是可以正常的运行，运行3、4个epoch之后，就出现了上面的错误提示，于是尝试把torch.argmax()中的dim修改成-1或者0，跑的通，但是所得结果不对。于是放弃AUC>threshold的时候去算AUC2的想法。

 

2.按照之前的想法，分别计算AUC和