Demo code Question

[AlderMen]

Hello,
I downloaded the data of the Burgers equation you provided and inserted it into your demo program to run it, but an error message appeared:

These are the information of Burgers equation:

Burgers Dataset has been loaded successfully!
X train shape: torch.Size([1000, 256, 1]) Y train shape: torch.Size([1000, 256])
X test shape: torch.Size([200, 256, 1]) Y test shape: torch.Size([200, 256])
The autoencoder type is MLP.
KNO2d model is completed.
Koopman Model has been compiled!
The Model Parameters Number is  526026

This is the code error message:
RuntimeError: mat1 and mat2 shapes cannot be multiplied (16384x1 and 10x32)

Could you help me solve this problem? Thanks a lot