运行是出现'Dimension out of range'

[zhengzelingjun]

(pytorch) E:\genshin_auto_fish>python fishing.py image -f yolox/exp/yolox_tiny_fish.py -c weights/best_tiny3.pth --conf 0.25 --nms 0.45 --tsize 640 --device cpu 2021-12-05 10:04:11.000 | INFO | main:main:91 - Args: Namespace(ckpt='weights/best_tiny3.pth', conf=0.25, demo='image', device='cpu', exp_file='yolox/exp/yolox_tiny_fish.py', experiment_name='yolox_tiny_fish', fp16=False, fuse=False, legacy=False, model_dir='./weights/fish_genshin_net.pth', n_actions=2, n_states=3, name=None, nms=0.45, path='./assets/dog.jpg', step_tick=12, trt=False, tsize=640) Traceback (most recent call last): File "fishing.py", line 195, in main(exp, args) File "fishing.py", line 101, in main logger.info("Model Summary: {}".format(get_model_info(model, exp.test_size))) File "E:\genshin_auto_fish\yolox\utils\model_utils.py", line 23, in get_model_info flops, params = profile(deepcopy(model), inputs=(img,), verbose=False) File "C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\thop\profile.py", line 204, in profile model(inputs) File "C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl return forward_call(input, *kwargs) File "E:\genshin_auto_fish\yolox\models\yolox.py", line 30, in forward fpn_outs = self.backbone(x) File "C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl return forward_call(input, **kwargs) File "E:\genshin_auto_fish\yolox\models\yolo_pafpn.py", line 99, in forward f_out0 = torch.cat([f_out0, x1], 1) # 512->1024/16 IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)