search

lu-group / deeponet-fno

DeepONet & FNO (with practical extensions)
Other
233 stars 49 forks source link

Run burgers equation error #10

Open rochaie opened 1 year ago

rochaie commented 1 year ago

RuntimeError: The size of tensor a (2) must match the size of tensor b (16) at non-singleton dimension 3; After removes optimizer 's param weight_decay, the model run normally,but training abnormal How should I modify it？

lululxvi commented 1 year ago

More details?

xiaoyuxie-vico commented 11 months ago

RuntimeError: The size of tensor a (2) must match the size of tensor b (16) at non-singleton dimension 3; After removes optimizer 's param weight_decay, the model run normally,but training abnormal How should I modify it？

Change the version of pytorch. The problem can be solved.

lululxvi commented 10 months ago

@xiaoyuxie-vico What is the PyTorch version required?

xiaoyuxie-vico commented 10 months ago

@xiaoyuxie-vico What is the PyTorch version required?

Pytorch1.13.0+cu116 works for me.

asjkckabca commented 2 months ago

RuntimeError: The size of tensor a (2) must match the size of tensor b (16) at non-singleton dimension 3; After removes optimizer 's param weight_decay, the model run normally,but training abnormal How should I modify it？

Have you solved the issue? Thank you！