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 提醒我维度不匹配，有人遇到这个问题吗