From 35768a843088a045c73ac55951e6d4fa2e3d60c1 Mon Sep 17 00:00:00 2001 From: martins0n <33594071+martins0n@users.noreply.github.com> Date: Thu, 18 Aug 2022 18:58:17 +0300 Subject: [PATCH 1/2] FIX: flags pytorch light trainer --- etna/models/nn/deepar.py | 1 - etna/models/nn/tft.py | 1 - 2 files changed, 2 deletions(-) diff --git a/etna/models/nn/deepar.py b/etna/models/nn/deepar.py index 74cb6698e..e7087bd60 100644 --- a/etna/models/nn/deepar.py +++ b/etna/models/nn/deepar.py @@ -156,7 +156,6 @@ def fit(self, ts: TSDataset) -> "DeepARModel": logger=tslogger.pl_loggers, max_epochs=self.max_epochs, gpus=self.gpus, - checkpoint_callback=False, gradient_clip_val=self.gradient_clip_val, ) trainer_kwargs.update(self.trainer_kwargs) diff --git a/etna/models/nn/tft.py b/etna/models/nn/tft.py index 427ad85f3..878607b7b 100644 --- a/etna/models/nn/tft.py +++ b/etna/models/nn/tft.py @@ -163,7 +163,6 @@ def fit(self, ts: TSDataset) -> "TFTModel": logger=tslogger.pl_loggers, max_epochs=self.max_epochs, gpus=self.gpus, - checkpoint_callback=False, gradient_clip_val=self.gradient_clip_val, ) trainer_kwargs.update(self.trainer_kwargs) From 14c6e9bd91d18752f8176ae7dc6974e76e738971 Mon Sep 17 00:00:00 2001 From: martins0n <33594071+martins0n@users.noreply.github.com> Date: Thu, 18 Aug 2022 19:01:03 +0300 Subject: [PATCH 2/2] FIX: changelog --- CHANGELOG.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index f6d561fb4..5e9296d09 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -39,7 +39,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 - - - Fix behavior of SARIMAXModel if simple_differencing=True is set ([#837](https://github.com/tinkoff-ai/etna/pull/837)) -- +- Fix deprecated pytorch lightning trainer flags ([#866](https://github.com/tinkoff-ai/etna/pull/866)) - - - ProphetModel doesn't work with cap and floor regressors ([#842](https://github.com/tinkoff-ai/etna/pull/842))