Run burgers equation error

[rochaie]

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]

More details?

[xiaoyuxie-vico]

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]

@xiaoyuxie-vico What is the PyTorch version required?

[xiaoyuxie-vico]

@xiaoyuxie-vico What is the PyTorch version required?

Pytorch1.13.0+cu116 works for me.

[asjkckabca]

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！