This commit is contained in:
Varuna Jayasiri
2020-09-10 18:08:55 +05:30
parent 92aca4be32
commit ee0e9728e9
2 changed files with 5 additions and 1 deletions

View File

@ -150,5 +150,5 @@ class RHN(Module):
# Stack the outputs and states
out = torch.stack(out)
s = torch.stack(s)
return out, s