Timegrad error on wiki data

[jiajinghu19]

I am running the Time-Grad-Electricity.ipynb notebook on wiki data and got an error below.

RuntimeError: input.size(-1) must be equal to input_size. Expected 28611, got 6006

Is it because wiki data has 9000+ dimension and train_grouper = MultivariateGrouper(max_target_dim=min(2000, int(dataset.metadata.feat_static_cat[0].cardinality))) cut it into 2000?

[kashif]

right depends on several things but yes that could be one reason... have a look at this example that calculates the input size based on the dataset: https://github.com/kashif/time_match/blob/main/Time-Grad-Solar.ipynb