Minor error

ari-dasci · Oct 31, 2023 · 3bd3867 · 3bd3867
1 parent d4462ec
commit 3bd3867
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -121,7 +121,7 @@ Installation updated for tensorflow, pytorch and newer versions of Python.
 - Removed:
 - Fixed:
 
-1.0.7.3 (31 Oct 2023)
+1.0.7.4 (31 Oct 2023)
 ======================
 **Description**
 

diff --git a/TSFEDL/models_pytorch.py b/TSFEDL/models_pytorch.py
@@ -70,8 +70,8 @@ def training_step(self, batch, batch_idx):
 
     def validation_step(self, batch, batch_idx):
         # this is the validation loop
-        src,tgt,y = batch
-        y_hat = self(src,tgt)
+        x,y = batch
+        y_hat = self(x)
         val_loss = self.loss(y_hat, y)
         self.log("val_loss", val_loss, prog_bar=True)
         return val_loss

diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
     name='TSFEDL',
 
     # Version
-    version='1.0.7.3',
+    version='1.0.7.4',
 
     # Description
     description='Time Series Spatio-Temporal Feature Extraction using Deep Learning',