-
Notifications
You must be signed in to change notification settings - Fork 33
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
When using the Multi-dimensional Balanced MSE, there is a memory error #22
Comments
What is your feature dimension? The loss requires |
oh,thank you. I used a dense layer to reduce dimensions to solve the memory issue. However, I encountered a new problem: an error occurred in the line of code that calls torch's MVN, reporting a RuntimeError: CUDA error: no kernel image is available for execution on the device. I look forward to your reply. |
Hi, I am not sure how to solve this issue, you may look for solutions in pytorch forums. |
I had solved this problem by reducing dimensions.Thank you. |
When using the Multi-dimensional Balanced MSE for pixel-wise loss,there is an error:not enough memory: you tried to allocate 9663676416 bytes. Buy new RAM!
The text was updated successfully, but these errors were encountered: