You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
TorchForecastingModel._setup_trainer() could be adjusted to not save the trainer to self but rather return a Trainer instance.
I have found that I use different trainer args for training versus inference. That being said, it is possible on model initialization to save the trainer args and create a trainer for fit() and predict() based on these with the option to override when calling fit() or predict().
@alexcolpitts96, the PR adjusts TorchForecastingModel._setup_trainer() but we cannot avoid storing the trainer as an attribute in the model object due to week reference errors (the trainer would get dropped after fit/predict calls if we don't store it -> PyTorch Lightning will lsoe the reference to the trainer).
No description provided.
The text was updated successfully, but these errors were encountered: