About train problem

[zyz-notebooks]

When I was training, the following problems occurred in the middle of my training. looking forward to your reply.

2020-09-08 09:50:59.346583 Iter: 2880 Loss: 0.171822 Wait until the dataprocesses to end Traceback (most recent call last): File "main.py", line 121, in main() File "main.py", line 111, in main train_net() File "/home/zyz/Project/3D-R2N2-PyTorch/lib/train_net.py", line 24, in func_wrapper return func(*args, **kwargs) File "/home/zyz/Project/3D-R2N2-PyTorch/lib/train_net.py", line 84, in train_net solver.train(train_loader, val_loader) File "/home/zyz/Project/3D-R2N2-PyTorch/lib/solver.py", line 117, in train loss = self.train_loss(batch_img, batch_voxel) File "/home/zyz/Project/3D-R2N2-PyTorch/lib/solver.py", line 65, in train_loss loss = self.net(x, y, test=False) File "/home/zyz/anaconda3/envs/3D-R2N2/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/zyz/Project/3D-R2N2-PyTorch/models/base_gru_net.py", line 76, in forward gru_out, update_gate = self.encoder(x[time], h, u, time) File "/home/zyz/anaconda3/envs/3D-R2N2/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/zyz/Project/3D-R2N2-PyTorch/models/res_gru_net.py", line 155, in forward t_x_s_update = self.t_x_s_update(rect7, h) File "/home/zyz/anaconda3/envs/3D-R2N2/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/zyz/Project/3D-R2N2-PyTorch/lib/layers.py", line 45, in forward out = self.fc_layer(fc7).view(*target_shape) + self.conv3d(h) + self.bias RuntimeError: The size of tensor a (5) must match the size of tensor b (12) at non-singleton dimension 0

[Wszzn]

I have met the same problem and later I solve with it. You only need to add "drop_last=True" to the data_loader. I hope it will be helpful. Thank u.