-
Notifications
You must be signed in to change notification settings - Fork 232
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
train question #63
Comments
我现在把batch-size改为8,输入图片改为256*256解决了这个错误,但是不得不说,LIP数据集太大了,还是很吃配置的 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
dear author,thanks your code. while I train LIP dataset,I occur a question.I use 4 GPU:2080ti,batch_size=8,num_worker=1,but found :"RuntimeError: CUDA out of memory. Tried to allocate 28.00 MiB (GPU 7; 10.76 GiB total capacity; 262.52 MiB already allocated; 22.12 MiB free; 312.00 MiB reserved in total by PyTorch)", can you give some advice to solve this question? Thanks
The text was updated successfully, but these errors were encountered: