diff options
| author | Clement Farabet <clement.farabet@gmail.com> | 2011-08-26 03:07:01 +0400 |
|---|---|---|
| committer | Clement Farabet <clement.farabet@gmail.com> | 2011-08-26 03:07:01 +0400 |
| commit | 2402831b4a641ecf65fc27cdd4a9551694b8d710 (patch) | |
| tree | 1de7e1e92b47b4e7bdc2ea46eaa9834ddaf74422 | |
| parent | 20ad9917d9bdea4fc0ce6d220198aa7e9ace9189 (diff) | |
Cleaned up useless special case in mapreduce bfgs.
| -rw-r--r-- | LBFGSOptimization.lua | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/LBFGSOptimization.lua b/LBFGSOptimization.lua index 4760cf2..0eff7b4 100644 --- a/LBFGSOptimization.lua +++ b/LBFGSOptimization.lua @@ -91,13 +91,8 @@ function LBFGS:forward_mapreduce(inputs, targets, options) self.parametersPT = {} self.gradParametersPT = {} for m = 1,self.parallelize do - if m == 1 then - modules[m] = self.module - criterions[m] = self.criterion - else - modules[m] = self.module:clone() - criterions[m] = self.criterion:clone() - end + modules[m] = self.module:clone() + criterions[m] = self.criterion:clone() self.parametersPT[m] = nnx.getParameters(modules[m]) self.gradParametersPT[m] = nnx.getGradParameters(modules[m]) end |