-
Notifications
You must be signed in to change notification settings - Fork 157
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
RuntimeError: each element in list of batch should be of equal size #96
Comments
can you show me more infos? |
Sure, I prepared a notebook which you can use to reproduce the error. Just note that I'm using a subset of the SynthText dataset due to disk storage limits on google colab, so there are several steps that run before your code that will download the subset and filter out only the corresponding data while maintaining the same shapes expected by a slightly modified version of your code. Besides, I tried running your code with the full dataset without any modifications on my behalf, it runs without displaying any progress for over an hour, so I'm assuming something is failing silently somewhere. |
any updates on this? |
@unsignedrant Have you solved this problem? I meet the same issue. |
@ziyeZzz No, I haven't and I reconsidered using craft for text detection because of the absence of implementation details in the paper as well as here on github, all of the existing implementations similar to this one are broken. |
@unsignedrant I just found that if set bs=1, the issue can pass. But I haven't figured out how to make bs>1 . |
When I use pytorch1.8.1,python3.7.9, batch_size>1, the code will report "each element in list of batch should be of equal size" If pytorch1.8.1,python3.7.9, batch_size=1, when training_step>1w(nearby), training_loss=nan. When I use pytorch 1.5.1, python 3.5, the code can work fine regardless the value of batch_size. Do you have any idea of this issue? @backtime92 |
delete the variable "confidences", the len of confidences will vary from image to image |
I get this error when I run
python trainSynth.py
with SynthText as datasetThe text was updated successfully, but these errors were encountered: