From 9f877f8730322a9b74a4df1bfb9d245cca968b61 Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 27 Dec 2018 22:02:53 -0800 Subject: [PATCH] update GRUD --- GRUD.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GRUD.py b/GRUD.py index 263ecc8..31c5f80 100644 --- a/GRUD.py +++ b/GRUD.py @@ -123,7 +123,7 @@ def step(self, x, h, mask, delta, x_mean): combined = torch.cat((x, h, mask), 1) z = F.sigmoid(self.zl(combined)) r = F.sigmoid(self.rl(combined)) - combined_r = torch.cat((x, r * Hidden_State, mask), 1) + combined_r = torch.cat((x, r * Hidden_State, mask), 1) h_tilde = F.tanh(self.hl(combined_r)) h = (1 - z) * h + z * h_tilde