通過學習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
相關文章
相關標籤/搜索