错误类型:
今天在跑自己的数据集+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属性删除即可。
删除前
删除后
删除后即可正常运行