RuntimeError: CUDA error: out of memory
wanghai111 opened this issue · 1 comments
wanghai111 commented
When the code runs for a while, it will have problems with out of memory.
luuuyi commented
@wanghai111 maybe you can try to decrease the batch_size