From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sat, 13 Oct 2018 13:40:19 +0000 (+0100)
Subject: [Minor] Always set number of threads when performing learn
X-Git-Tag: 1.8.1~30
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=fe026097b171d411c376fb0afb5d11b3354676fc;p=thirdparty%2Frspamd.git

[Minor] Always set number of threads when performing learn
---

diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua
index 0221600832..2d659474eb 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()