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

Published 20 hours ago • verified publisher icon dee1024

训练的时候报错

Open zlp19860220 opened this issue 7 years ago • 1 comments

File "C:\Users\leping.zhao\AppData\Local\Programs\Python\Python36\lib\site-pac kages\torch\utils\data\dataloader.py", line 165, in default_collate return torch.stack(batch, 0, out=out) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimensio n 0. Got 160 and 164 in dimension 3 at c:\new-builder_2\win-wheel\pytorch\aten\s rc\th\generic/THTensorMath.cpp:3616

当验证码只要30个的时候没有报错，但是把数量改成1W或者5W生成验证后，然后训练的时候就报这个错误了，麻烦看下怎么改

zlp19860220 avatar Nov 16 '18 07:11 zlp19860220

我也是求解决

SFHhhh avatar Sep 21 '19 09:09 SFHhhh