search

ayumiymk / aster.pytorch

ASTER in Pytorch
MIT License
665 stars 169 forks source link

CUDA out of memory problem #47

Closed Pgwsunny closed 4 years ago

Pgwsunny commented 4 years ago

CUDA out of memory 训练模型时遇到以下问题。我想知道作者是否正在将所有数据加载到内存中吗？还有其他解决方法吗？非常感谢你。

12 12

ayumiymk commented 4 years ago

Maybe you can reduce the batchsize or use more gpus.