-
Notifications
You must be signed in to change notification settings - Fork 260
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
训练part_A数据集,batch size=1,loss震荡很厉害,有什么解决办法吗 #83
Comments
把batch size调大 |
我冒昧揣测一下, 你用bsize=1是因为显存不够用吧? 如果是的话, 大胆一点, 把vgg层数和每层通道数降一降. 或许能顺便解决 loss震荡 的次生灾害. |
Setting the batch size to 1 is because the varying dimensions of input images. If all images had the same dimensions, we could increase the batch size. However, since the dimensions of the images are different, the batch size has to be set to 1. This is because we cannot stack images with different sizes into a single batch. |
No description provided.
The text was updated successfully, but these errors were encountered: