跑您的模型的时候遇到的一些问题

[LihuiNb]

您好，非常欣赏您的idea，所以尝试跑一下您的分类模型。 我下载了ImageNet2012数据集之后，尝试启动您的模型，遇到了以下问题，想请教一下是否我的哪些设置出错了？

错误信息如下： Traceback (most recent call last): File "main.py", line 643, in main() File "main.py", line 389, in main avg_train_time = train(train_loader, model, criterion, optimizer, epoch, logger, scheduler) File "main.py", line 471, in train prec1, prec5 = accuracy(output.data, target, topk=(1, 5)) File "main.py", line 631, in accuracy 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.

[cfzd]

@LihuiNb 您好，我没有遇到过这种问题，不过看报错信息的话可以这样修改： correct_k = correct[:k].view(-1).float().sum(0, keepdim=True) 改为 correct_k = correct[:k].contiguous().view(-1).float().sum(0, keepdim=True)

[LihuiNb]

谢谢，您非常专业，问题已解决。