刚开始学习动作识别相关知识,现在在TSM中用BNInception作为主干网络尝试训练时遇见这个问题,求指导!
Traceback (most recent call last):
File "main.py", line 378, in
main()
File "main.py", line 194, in main
train(train_loader, model, criterion, optimizer, epoch, log_training, tf_writer)
File "main.py", line 244, in train
output = model(input_var)
File "/home/miao/software/anaconda3/envs/mmaction2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/miao/software/anaconda3/envs/mmaction2/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 165, in forward
return self.module(*inputs[0], **kwargs[0])
File "/home/miao/software/anaconda3/envs/mmaction2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/miao/software/TSM/ops/models.py", line 270, in forward
base_out = self.base_model(input.view((-1, sample_len) + input.size()[-2:]))
File "/home/miao/software/anaconda3/envs/mmaction2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/miao/software/TSM/archs/bn_inception.py", line 44, in forward
x = self.features(x)
File "/home/miao/software/TSM/archs/bn_inception.py", line 52, in features
pool2_3x3_s2_out = self._temporal_forward_wrap(self._block_2, 1)(pool1_3x3_s2_out)
File "/home/miao/software/TSM/archs/bn_inception.py", line 100, in wrapped_func
return F.relu(x + layer_func(x_shift))
RuntimeError: The size of tensor a (56) must match the size of tensor b (28) at non-singleton dimension 3