pytorch-captcha-recognition icon indicating copy to clipboard operation
pytorch-captcha-recognition copied to clipboard

这是什么问题啊

Open SFHhhh opened this issue 5 years ago • 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

SFHhhh avatar Sep 21 '19 09:09 SFHhhh