about RuntimeError: CUDA out of memory

coolbay / VSGN

MIT License

25 stars 0 forks source link

about RuntimeError: CUDA out of memory #3

Open Beatricehuihuihui opened 1 year ago

Beatricehuihuihui commented 1 year ago

1684226240511 Hi, thx for releasing the code. I just reproduce it, but got an runtime error——cuda out of memory. I decrease the batch size from 32 to 16, 8, 4, 2, but the error always happens. Do you know why?