Self_Explaining_Structures_Improve_NLP_Models icon indicating copy to clipboard operation
Self_Explaining_Structures_Improve_NLP_Models copied to clipboard

Published 20 hours ago • verified publisher icon ShannonAI

CUDA out of memory

Open amiaynara opened this issue 3 years ago • 2 comments

Tried to allocate 20.00 MiB (GPU 0; 1.96 GiB total capacity; 1.24 GiB already allocated; 2.88 MiB free; 1.25 GiB reserved in total by PyTorch)

amiaynara avatar May 20 '21 18:05 amiaynara

maybe you can use batch accumulate to decrease the batch size.

zijunsun avatar Jun 15 '21 06:06 zijunsun

Tried to allocate 20.00 MiB (GPU 0; 1.96 GiB total capacity; 1.24 GiB already allocated; 2.88 MiB free; 1.25 GiB reserved in total by PyTorch)

你是在window上跑的嘛

huanghao-att avatar Jul 04 '21 03:07 huanghao-att