RuntimeError: $ Torch: not enough memory: you tried to allocate 2GB
outputs = net()在上面这行遇到内存不够的问题,一定要考虑net()里面的batch size是不是太大了。即使你是在测试,但如果你是直接把整个数据集输入到net()中,一般来说内存是不够用的,你必须把batch size减少到正常水平,比如256。参考:https://github.com/SeanNaren/deepspeech.pytorch/issues/150...
原创
2020-02-20 08:49:33 ·
3173 阅读 ·
0 评论