Error in runing train_MBN2_mod.py

[Liuhailong2016]

Hi.
I used CASIA 2.0 dataset and train MBN2 model, in running train_MBN2_mod.py, the system raised the error as below:

---

Train on epoch 1
0%| | 0/19 [00:00<?, ?it/s]
Traceback (most recent call last):
File "train_MBN2_mod.py", line 163, in
loss_train, acc_train, time_train = train_on_epoch(
File "/home/rr-ubuntu/pythonproject/Image-Forgery-using-Deep-Learning/utils/learning.py", line 32, in train_on_epoch
logits = model(X)
File "/home/rr-ubuntu/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/rr-ubuntu/pythonproject/Image-Forgery-using-Deep-Learning/utils/models.py", line 217, in forward
out = self.linear(out)
File "/home/rr-ubuntu/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/rr-ubuntu/.local/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 93, in forward
return F.linear(input, self.weight, self.bias)
File "/home/rr-ubuntu/.local/lib/python3.8/site-packages/torch/nn/functional.py", line 1690, in linear
ret = torch.addmm(bias, input, weight.t())
RuntimeError: mat1 dim 1 must match mat2 dim 0

How can I do for solving this problem?

[Liuhailong2016]

I have solved this problem.
The reason is the photos from CASIA 2.0 dataset is 384256, but the input of the netural network is 6464, so we should resize the image before training as below: