diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2018-10-13 14:40:19 +0100 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2018-10-13 14:40:19 +0100 |
| commit | fe026097b171d411c376fb0afb5d11b3354676fc (patch) | |
| tree | 4ee49324fbb4616d5d6e00e9af9209a1a91d70c7 /src/plugins | |
| parent | 7a58e9057c978442d55069d4eef230df615cd714 (diff) | |
| download | rspamd-fe026097b171d411c376fb0afb5d11b3354676fc.tar.gz rspamd-fe026097b171d411c376fb0afb5d11b3354676fc.zip | |
[Minor] Always set number of threads when performing learn
Diffstat (limited to 'src/plugins')
| -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() |