错误类型:
今天在跑自己的数据集+AlexNet时一直报错,错误如下
错误提示:
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.FloatTensor [544, 768]], which is output 0 of ViewBackward, is at version 1; expected version 0 instead.
Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).
解决方案:
把网络模型线性分类器中Dropout层的inplace=True属性删除即可。

删除前

删除后
删除后即可正常运行
在运行自定义数据集上的AlexNet模型时遇到RuntimeError,错误涉及到了梯度计算中的变量版本不匹配。问题出在Dropout层的inplace属性设置为True。删除这一属性后,模型可以正常运行。解决方案是将线性分类器中Dropout层的inplace属性设为False。
7742

被折叠的 条评论
为什么被折叠?



