huawei-noah / vega

AutoML tools chain
http://www.noahlab.com.hk/opensource/vega/
Other
842 stars 175 forks source link

Error when running prune_ea #222

Open L-ED opened 2 years ago

L-ED commented 2 years ago

ERROR:root:Failed to run worker, id: 1, message: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [256, 64, 8, 8]], which is output 0 of ReluBackward0, 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

zhangjiajin commented 2 years ago

@L-ED What is the version of pytorch on your environment?

L-ED commented 2 years ago

torch 1.10

zhangjiajin commented 2 years ago

@L-ED

We tested this version of pytorch and reproduced the problem. We're working on that.

Please try an earlier version of pytorch.

L-ED commented 2 years ago

谢谢你帮助我.