PAN-PyTorch RuntimeError: shape '[-1, 3, 224, 224]' is invalid for input of size 12288

RuntimeError: shape '[-1, 3, 224, 224]' is invalid for input of size 12288

Open Chunssu opened this issue 2 years ago • 5 comments

在测试somethingv2数据集的时候出现如下问题，在En.sh,Full.sh,Lite.sh三个测试脚本上都会有如下情况。 I found RuntimeError when I running somethingv2 test scripts.

Traceback (most recent call last): File "C:\Users\ganjunsi\Downloads\PAN\test_models.py", line 289, in rst = eval_video((i, data, label), net, n_seg, modality) File "C:\Users\ganjunsi\Downloads\PAN\test_models.py", line 128, in eval_video rst = net(data_in) File "C:\Users\ganjunsi.conda\envs\gjs-pan\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\ganjunsi.conda\envs\gjs-pan\lib\site-packages\torch\nn\parallel\data_parallel.py", line 166, in forward return self.module(*inputs[0], **kwargs[0]) File "C:\Users\ganjunsi.conda\envs\gjs-pan\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\ganjunsi\Downloads\PAN\ops\models.py", line 290, in forward PA = self.PA(input) File "C:\Users\ganjunsi.conda\envs\gjs-pan\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl return forward_call(*input, *kwargs) File "C:\Users\ganjunsi\Downloads\PAN\ops\PAN_modules.py", line 28, in forward PA = d.view(-1, 1(self.n_length-1), h, w) RuntimeError: shape '[-1, 3, 224, 224]' is invalid for input of size 12288