0

我是这个领域的新手,并试图通过下面的 Github 链接学习。但是,我遇到了运行时错误。尽管将批量大小和梯度累积调整为更小的值,并清除了缓存,但运行时错误仍然存​​在。

任何人都可以分享对此的任何见解吗?非常感谢。

https://github.com/nlptown/nlp-notebooks/blob/master/Text%20classification%20with%20BERT%20in%20PyTorch.ipynb

运行时错误:CUDA 内存不足。尝试分配 1024.00 KiB(GPU 0;4.00 GiB 总容量;3.09 GiB 已分配;528.00 KiB 空闲;32.48 MiB 缓存)

4

0 回答 0