CUDA out of memory. (解决了)
报错
Traceback (most recent call last):
File "plastic_train.py", line 243, in <module>
y, hebb = plasticnet.module(Variable(inputs[numstep], requires_grad=False), y, hebb)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py"
原创
2020-08-28 19:24:59 ·
2866 阅读 ·
0 评论