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