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 ·
2686 阅读 ·
0 评论