chengtan9907 / OpenSTL

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
https://openstl.readthedocs.io/en/latest/
Apache License 2.0
730 stars 113 forks source link

Error when using trained ConvLSTM with new dataset #157

Open BlackRab opened 2 weeks ago

BlackRab commented 2 weeks ago

Hello. I'm a newer to OpenSTL. I've trained a ConvLSTM, the shape of the input is (n, 12, 1, 64, 64), where n is the number of samples, and the shape of output is (n, 24, 1, 64, 64). But when I use the trained model to predict on a new dataset, I've met an error: TypeError: forward() missing 1 required positional argument: 'batch_y'

The code for prediction is as follows:

predj = trainer.predict(model, dipti)

where dipti is a dataloader with data shape of (1, 12, 1 64, 64).

How can I solve this error? Thanks!