File "/home/nkw/code/hmp2g-master/ALGORITHM/common/hyper_net.py", line 49, in forward
hidden = F.relu(torch.matmul(x_, w1) + b1) # b * t, 1, emb
RuntimeError: Expected size for first two dimensions of batch2 tensor to be: [10, 360] but got: [10, 264].
似乎是报错计算出的维度不正确,请问如何解决?
File "/home/nkw/code/hmp2g-master/ALGORITHM/common/hyper_net.py", line 49, in forward
hidden = F.relu(torch.matmul(x_, w1) + b1) # b * t, 1, emb
RuntimeError: Expected size for first two dimensions of batch2 tensor to be: [10, 360] but got: [10, 264].
似乎是报错计算出的维度不正确,请问如何解决?