pytorch train error

[smowtion]

Hello I have captcha and want train it

but error

result = self.forward(*input, **kwargs) File "train.py", line 38, in forward x = x.view(-1, 48 * 6 * 27) # flatten here RuntimeError: shape '[-1, 7776]' is invalid for input of size 475968

my data has : 45,595 file CHARS = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'K', 'M', 'N','L','S','I', 'P', 'R', 'T', 'U', 'V', 'W', 'X', 'Y','1','2','3','4','5','6','7','8','9'] And my captcha lenght: 6 character How I can config file train.py correct thank

[cyy53589]

Maybe you need to resize pic to 120 * 36