From 1eb41d87b476d283c0ef13efb6176f0e55d48836 Mon Sep 17 00:00:00 2001 From: Jan Fidor Date: Sun, 23 Jul 2023 17:03:06 +0200 Subject: [PATCH] add bug fix after merging --- darts/models/forecasting/transformer_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/darts/models/forecasting/transformer_model.py b/darts/models/forecasting/transformer_model.py index d17082cb08..6d494cfdb8 100644 --- a/darts/models/forecasting/transformer_model.py +++ b/darts/models/forecasting/transformer_model.py @@ -349,6 +349,7 @@ def _prediction_step(self, src: torch.Tensor, tgt: torch.Tensor): # Allow teacher forcing def training_step(self, train_batch, batch_idx) -> torch.Tensor: """performs the training step""" + train_batch = list(train_batch) future_targets = train_batch[-1] train_batch.append(future_targets) return super().training_step(train_batch, batch_idx)