-
Notifications
You must be signed in to change notification settings - Fork 63
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
CUDA out of memory. #7
Comments
You can reduce the content size and style size in the training process. But I am not sure if it will affect the stylization effect. |
can you test it now? can we communicate? |
I can't test successfully and i don't know why. |
my wechat is c319186779.Can we communicate more? |
Hallo, can you test successfully now? How to slove the error tuple object has no attribute cpu |
for content_path in content_paths:
You can copy this code to replace corresponding code on test.py |
Did you successfully test it?I editted the code as yours but in vain, may you share your environment? |
Maybe you can try the following code:
|
Thank u so much!It works perfectly,and I found a mistake in the version of cuda,the code can only run on the cuda 10.x or lower. |
Have you solved your problem? I tried to run this code in RTX3080 with 10G memory and got "CUDA OUT OF MEMORY". I can only run this code when the batch size is 1. Once the batch size is greater than 1, even if iters is 1, it will report an error. |
收到了,谢谢
|
I try to train this code on 23090, but it still replys that CUDA out of memory. In your paper, you say that your model is trained on 2 Tesla V100 and 2 GeForce RTX 3090 GPUs spending about one day. I want to know whether this code can be trained on 23090. Than you for your contribution.
The text was updated successfully, but these errors were encountered: