通过学习BATCH_SIZE 和epoch,解决内存不足的报错

训练时出现了这种错误: RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 4.22 GiB already allocated; 4.91 MiB free; 85.24 MiB cached) 运行时错误:CUDA内存不足。尝试分配20.00 MiB(GPU
相关文章
相关标签/搜索