记录报错过程以及解决方法
1、loss保持不变,始终稳定在0.208
1.1在计算loss时数据维数不匹配
from torch import nn
loss = nn.CrossEntropyLoss(outputs, labels)
经过查看可发现outputs.shape为torch.Size([64, 64, 10])
而labels.shape为torch.Size([64, 10])
1.1.1使用view时报错:
RuntimeError: view size is not compatible with input tensor’s size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(…) instead.
更改:
将变量先转为cont