zalandoresearch / pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend
MIT License
1.24k stars 191 forks source link

TypeError: `model` must be a `LightningModule` or `torch._dynamo.OptimizedModule`, got `TimeGradLightningModule` #157

Open bhuwanKarki opened 9 months ago

bhuwanKarki commented 9 months ago

when installing the pyyorch-ts and initiating the model i get the following error any idea to solve it