[FIX] Fixes 948 multivariate predict/val issues when n_series > 1024

[elephaint]

By default, valid_batch_size = 1024 in BaseModel which gives an issue with BaseMultivariate if n_series > 1024. This fix simply sets the valid_batch_size =n_series, similar to how batch_size is implemented in BaseMultivariate.

Summary:

• Fixes TSMixerx model batch size mismatch if number of unique_ids > 1024 #948
• Added unit test to TSMixerx model that fails without the fix (unit test should also work for other multivariate models, I only included it for TSMixerx though, since it's more related to BaseMultivariate than a specific model)

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB