Please help me how to rectify this:MisconfigurationException: The provided lr scheduler StepLR doesn't follow PyTorch's LRScheduler API. You should override the LightningModule.lr_scheduler_step hook with your own logic if you are using a custom LR scheduler.

[Harish11999]

[AzulGarza]

hey @Harish11999! Thanks for using neuralforecast. The problem is related to the latest version of torch (v2.0.0). You can solve the issue by installing the latest version of neuralforecast from the main branch. Here's the line:

pip install "git+https://github.com/nixtla/neuralforecast.git@main"

We are preparing an official release to pip soon. :)

[Harish11999]

I'm getting error again. The models will run but while we use ".fit", there is an error. I tried installing pytorch 1.6.5 but it doesn't exist.

[kyr0]

@Harish11999 You can pin to 1.3.0, then it will work in colab if you're not using the latest features added in 1.4.0+.

!pip install neuralforecast==1.3.0

[Harish11999]

Thank you. Now its fine