dee1024/pytorch-captcha-recognition

这是什么问题啊

SFHhhh opened this issue · 0 comments

Traceback (most recent call last):
File "/Users/haiyuan/Downloads/pytorch-captcha-recognition-master-2/captcha_train.py", line 43, in
main()
File "/Users/haiyuan/Downloads/pytorch-captcha-recognition-master-2/captcha_train.py", line 23, in main
for i, (images, labels) in enumerate(train_dataloader):
File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 346, in next
data = self.dataset_fetcher.fetch(index) # may raise StopIteration
File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 47, in fetch
return self.collate_fn(data)
File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 80, in default_collate
return [default_collate(samples) for samples in transposed]
File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 80, in
return [default_collate(samples) for samples in transposed]
File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/torch/utils/data/_utils/collate.py", line 56, in default_collate
return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 160 and 164 in dimension 3 at ../aten/src/TH/generic/THTensor.cpp:689