outputs = net()
在上面这行遇到内存不够的问题,一定要考虑net()里面的batch size是不是太大了。
即使你是在测试,但如果你是直接把整个数据集输入到net()中,一般来说内存是不够用的,你必须把batch size减少到正常水平,比如256。
参考:https://github.com/SeanNaren/deepspeech.pytorch/issues/150
outputs = net()
在上面这行遇到内存不够的问题,一定要考虑net()里面的batch size是不是太大了。
即使你是在测试,但如果你是直接把整个数据集输入到net()中,一般来说内存是不够用的,你必须把batch size减少到正常水平,比如256。
参考:https://github.com/SeanNaren/deepspeech.pytorch/issues/150