在运行自定义数据集上的AlexNet模型时遇到RuntimeError，错误涉及到了梯度计算中的变量版本不匹配。问题出在Dropout层的inplace属性设置为True。删除这一属性后，模型可以正常运行。解决方案是将线性分类器中Dropout层的inplace属性设为False。
错误类型：
今天在跑自己的数据集+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属性删除即可。
删除前
删除后
删除后即可正常运行
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
