search

lu-group / deeponet-fno

DeepONet & FNO (with practical extensions)
Other
202 stars 41 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 8 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 7 months ago

@xiaoyuxie-vico What is the PyTorch version required?

xiaoyuxie-vico commented 7 months ago

@xiaoyuxie-vico What is the PyTorch version required?

Pytorch1.13.0+cu116 works for me.