-
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
len() of a 0-d tensor #4
Comments
How about to change I can fix this issue by this in pytorch==1.0. |
|
Can you tell me the size of your gpu memory please? |
it is 12G,and you can reduce the number of picture in your dataset |
no no no , when I loaded the model , more than 5G memory is used. When the single picture is large than 10001000, it reports OOM! So I have to resize the image to 720720, but the result seems too bad~ |
If you are Chinese, can we talk on wechat ? |
I resize the image to 128*128,maybe it's ok. |
OK,I think I have found where I was wrong,if I have a good message,I will rely soon~ |
add a try...except block or judge if order.dim()==0,if so, i = order.item(), |
Hi, Thanks for releasing this code
I try to train the model on my dataset
but sometimes when I run the visualize notebook I get an error:
any idea why ?
The text was updated successfully, but these errors were encountered: