Traceback (most recent call last):
File “train.py”, line 75, in
train(model, trainloader, criterion, optimizer)
File “/home/bder73002/wln/HufuNet-master/utils.py”, line 161, in train
err1, err5 = get_error(output.detach(), target, topk=(1, 5))
File “/home/bder73002/wln/HufuNet-master/utils.py”, line 121, in get_error
correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
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.
根据上面报错我们可以知道是 File “/home/bder73002/wln/HufuNet-master/utils.py”, line 121, in get_error这里报错
然后百度我看到是因为Tensor资源没有整合报错,要在view之前加上 contiguous()
correct_k = correct[:k].contiguous().view(-1).float().sum(0, keepdim=True)
成功解决问题。