daviddoji/Fastbook_Julia
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed incorrect method call - torch.stack should be torch.cat, and fixed incorrect variable name (#338)

Browse Source 
Co-authored-by: Matus <test@test.com>
This commit is contained in:
Matus-Dubrava 2020-11-29 15:06:13 +01:00 committed by GitHub
parent b8b6b02aa4
commit c2fc05e063
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
12_nlp_dive.ipynb
View File

@ -1614,14 +1614,14 @@
"\n", "\n",
" def forward(self, input, state):\n", " def forward(self, input, state):\n",
" h,c = state\n", " h,c = state\n",
" h = torch.stack([h, input], dim=1)\n", " h = torch.cat([h, input], dim=1)\n",
" forget = torch.sigmoid(self.forget_gate(h))\n", " forget = torch.sigmoid(self.forget_gate(h))\n",
" c = c * forget\n", " c = c * forget\n",
" inp = torch.sigmoid(self.input_gate(h))\n", " inp = torch.sigmoid(self.input_gate(h))\n",
" cell = torch.tanh(self.cell_gate(h))\n", " cell = torch.tanh(self.cell_gate(h))\n",
" c = c + inp * cell\n", " c = c + inp * cell\n",
" out = torch.sigmoid(self.output_gate(h))\n", " out = torch.sigmoid(self.output_gate(h))\n",
" h = outgate * torch.tanh(c)\n", " h = out * torch.tanh(c)\n",
" return h, (h,c)" " return h, (h,c)"
] ]
}, },
@ -2362,4 +2362,4 @@
}, },
"nbformat": 4, "nbformat": 4,
"nbformat_minor": 2 "nbformat_minor": 2
} }