diff options
| author | Natalia Gimelshein <ngimelshein@nvidia.com> | 2016-04-25 19:14:02 +0300 |
|---|---|---|
| committer | Natalia Gimelshein <ngimelshein@nvidia.com> | 2016-04-25 19:14:02 +0300 |
| commit | d764ff20deda9039ce5885db4ed3808db659a8cc (patch) | |
| tree | d81ef7a5c4001174301ceb472af3c16a00951104 /DataParallelTable.lua | |
| parent | 6c3c3f3c5e21c9ad1a81507500d6d3113379691d (diff) | |
fix clearState for DataParallelTable
Diffstat (limited to 'DataParallelTable.lua')
| -rw-r--r-- | DataParallelTable.lua | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/DataParallelTable.lua b/DataParallelTable.lua index 81e5fe4..fd27f0d 100644 --- a/DataParallelTable.lua +++ b/DataParallelTable.lua @@ -155,6 +155,14 @@ function DataParallelTable:evaluate() parent.evaluate(self) end +function DataParallelTable:clearState() + self.impl:exec(function(module) + module:clearState() + end) + parent.clearState(self) +end + + function DataParallelTable:updateOutput(input) if self.flattenParams and not hasFlattenedParmeters(self) then self:flattenParameters() |