About train

[frenshman]

我换好SIDD数据集后运行 File "D:\Anaconda3\envs\Uformer\lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Anaconda3\envs\Uformer\lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "C:\Users\weixi\Desktop\去噪\Uformer-main\model.py", line 954, in forward
x = x.reshape(B, H, -1, C)
RuntimeError: shape '[15, 489, -1, 32]' is invalid for input of size 115200000
提醒我维度不匹配，有人遇到这个问题吗