Demo code running issue

[linanzhang]

Hello,

I played around with the parameters in the demo code and got a RuntimeError. When uploading the dataset and building the network, I changed t_in and t_out:

train_loader, test_loader = kp.data.navier_stokes(data_path, batch_size = 10, T_in = t_in, T_out = t_out, type = "1e-3", sub = 1)

koopman_model = kp.model.koopman(backbone = "KNO2d", autoencoder = "MLP", o = o, m = m, r = r, t_in = t_in, device = device)

An error message promts when calling koopman_model.train(), saying

RuntimeError: The size of tensor a (163840) must match the size of tensor b (122880) at non-singleton dimension 1

The error message is the same regarless of the value of t_in and t_out. What are the requirements on t_in and t_out?

Thank you.