Hi，When I tested, the following error occurred. May I ask how the code should be modified?
File "E:\PycharmProjects\CFEN-ViT-Dehazing-main\models\ipt.py", line 634, in forward
x = self.encoder(x+pos)
RuntimeError: The size of tensor a (1024) must match the size of tensor b (256) at non-singleton dimension 0
Thx
Hi，When I tested, the following error occurred. May I ask how the code should be modified? File "E:\PycharmProjects\CFEN-ViT-Dehazing-main\models\ipt.py", line 634, in forward x = self.encoder(x+pos) RuntimeError: The size of tensor a (1024) must match the size of tensor b (256) at non-singleton dimension 0 Thx