Open Nutingnon opened 1 week ago
Hi. We do not find this issue during our experiment. Could you please provide us with the dataset and script you used to help us fix the bug?
Sure, can you please give me an email address?
wenghk22@mails.tsinghua.edu.cn This is my email address. Thank you for your issue and discussion.
I was running experiment with FEDFormer, however, I met error in the
FourierCorrelation.py
at this line:The error message is:
RuntimeError: einsum(): subscript h has size 8 for operand 1 which does not broadcast with previously seen size 4
I print the following information before the above code as this:
The result is:
torch.Size([128, 4, 4]) torch.Size([8, 2, 2, 32]) torch.Size([8, 2, 2, 32]) torch.Size([8, 2, 2])
Then, I further double check the tensor information in the
compl_mul1d
function at the following linesand I get :
here: x size: torch.Size([128, 4, 4]) torch.Size([8, 2, 2])
I guess there is a bug in the implementation of FourierCorrelation.py file, I can successfully run my training data through other models such as AutoFormer, iTransformer, Informer, Mamba through your implementations.
Thank you for your great work and look forward to your answer.