Thesis/pytorch-dnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug: clamp the controller output before making it interface vector

Browse Source
This commit is contained in:
ixaxaar 2017-12-20 14:24:42 +05:30
parent 1abd4c4ba4
commit 08bd220852
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
dnc/dnc.py
View File

@ -178,14 +178,15 @@ class DNC(nn.Module):
input, chx = self.rnns[layer](input.unsqueeze(1), chx) input, chx = self.rnns[layer](input.unsqueeze(1), chx)
input = input.squeeze(1) input = input.squeeze(1)
# the interface vector
ξ = input
# clip the controller output # clip the controller output
if self.clip != 0: if self.clip != 0:
output = T.clamp(input, -self.clip, self.clip) output = T.clamp(input, -self.clip, self.clip)
else: else:
output = input output = input
# the interface vector
ξ = output
# pass through memory # pass through memory
if pass_through_memory: if pass_through_memory:
if self.share_memory: if self.share_memory: