维度不匹配

[Lan2Kailee]

首先谢谢您的代码，我在运行过程中出现了这样的问题，请问是什么原因呢？ PS D:\centernet\centernet> & D:/Anaconda3/envs/pytorch/python.exe d:/centernet/centernet/train.py Load weights model_data/res2net50_v1b_26w_4s-3cf99910.pth. Start Train Epoch 1/50: 0%| | 0/1034 [00:13<?, ?it/s<class 'dict'>] Traceback (most recent call last): File "d:/centernet/centernet/train.py", line 215, in epoch_step, epoch_step_val, gen, gen_val, end_epoch, Cuda, backbone) File "d:\centernet\centernet\utils\utils_fit.py", line 33, in fit_one_epoch hm, wh, offset = model_train(batch_images) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, kwargs) File "d:\centernet\centernet\nets\centernet.py", line 31, in forward feat = self.backbone(x) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, *kwargs) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(input, kwargs) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input) File "D:\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "d:\centernet\centernet\nets\resnet50.py", line 81, in forward sp = sp + spx[i] RuntimeError: The size of tensor a (64) must match the size of tensor b (128) at non-singleton dimension 3

[kirkzZ]

你好，我也遇到了这个问题，你这个问题解决了吗

[gasvn]

https://github.com/Res2Net/Res2Net-PretrainedModels/issues/68