diff options
Diffstat (limited to 'src/plugins/lua/neural.lua')
| -rw-r--r-- | src/plugins/lua/neural.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua index 022160083..2d659474e 100644 --- a/src/plugins/lua/neural.lua +++ b/src/plugins/lua/neural.lua @@ -659,7 +659,7 @@ local function train_ann(rule, _, ev_base, elt, worker) dataset.size = function() return #dataset end local function train_torch() - if rule.train.learn_threads > 1 then + if rule.train.learn_threads then torch.setnumthreads(rule.train.learn_threads) end local criterion = nn.MSECriterion() |