diff options
Diffstat (limited to 'LookupTable.lua')
| -rw-r--r-- | LookupTable.lua | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/LookupTable.lua b/LookupTable.lua index 7dfda7a..671783f 100644 --- a/LookupTable.lua +++ b/LookupTable.lua @@ -29,9 +29,13 @@ end function LookupTable:reset(stdv) stdv = stdv or 1 - self.weight:apply(function() - return torch.normal(0, stdv) - end) + if nn.oldSeed then + self.weight:apply(function() + return torch.normal(0, stdv) + end) + else + self.weight:normal(0, stdv) + end end function LookupTable:updateOutput(input) |