From 974f6fe5609b3edd5f305550c2d317c9d6531b82 Mon Sep 17 00:00:00 2001
From: Jose <34888496+Jerry-Master@users.noreply.github.com>
Date: Tue, 29 Aug 2023 14:36:26 +0200
Subject: [PATCH] GRU fix

---
 model/decoder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model/decoder.py b/model/decoder.py
index 7307435..99df84f 100644
--- a/model/decoder.py
+++ b/model/decoder.py
@@ -168,8 +168,8 @@ class ConvGRU(nn.Module):
     def forward_single_frame(self, x, h):
         r, z = self.ih(torch.cat([x, h], dim=1)).split(self.channels, dim=1)
         c = self.hh(torch.cat([x, r * h], dim=1))
-        h = (1 - z) * h + z * c
-        return h, h
+        h = (1 - z) * c + z * h
+        return c, h
     
     def forward_time_series(self, x, h):
         o = []