dataset shape not match in AVS

[zsevenj]

I try to run the code of avs, but the following error occurs:
XXXXXXXXXXXXXXXXX
File "../LAVISH/AVS/avs_scripts/avs_s4/model/PVT_AVSModel.py", line 492, in forward
x = self.swin.norm(x)
XXXXXXXXXXXXXXXXX
RuntimeError: Given normalized_shape=[1536], expected input with shape [*, 1536], but got input of size[5, 3, 192, 192]

[fyyCS]

May I ask if you've solved the issue?

[GenjiB]

I'll fix it recently.

[ShareWeb123]

Thank you for your excellent work. I've recently encountered a similar problem with debugging code.
File "../LAVISH/AVS/avs_scripts/avs_s4/model/PVT_AVSModel.py", line 492, in forward
x = self.swin.norm(x)
......
RuntimeError: Given normalized_shape=[1536], expected input with shape [*, 1536], but got input of size[20, 3, 192, 192]

[GenjiB]

Most of the bugs come from the version of timm. please use the exact version of timm and single gpu

[ShareWeb123]

Thank you very much for your reply. I installed timm==0.6.7 and set up single GPU, but it still prompts the same error "RuntimeError:Given normalized_shape=[1536],expected input with shape[*,1536],but got input of size[10,3,192,192]", the error occurs at line 492 x=self.swin.norm(x) in PVT_AVSModel.py. May I ask where else the problem could be caused? Looking forward to hearing from you, thanks again. At 2024-03-19 00:25:00, "GenjiB" ***@***.***> wrote: Most of the bugs come from the version of timm. please use the exact version of timm and single gpu — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: ***@***.***>