diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2017-09-17 10:04:59 +0100 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2017-09-17 10:04:59 +0100 |
| commit | 609d4f862202c43241fdab06b0795db6281ca3f7 (patch) | |
| tree | 22b838fc9f19772242d919a668f97b2b068d02c8 /lualib | |
| parent | 8308d6a6776ad73b8c0e61fcfdb55007434621cf (diff) | |
| download | rspamd-609d4f862202c43241fdab06b0795db6281ca3f7.tar.gz rspamd-609d4f862202c43241fdab06b0795db6281ca3f7.zip | |
[Feature] Allow to specify number of threads for ANN learning
Diffstat (limited to 'lualib')
| -rw-r--r-- | lualib/lua_nn.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lualib/lua_nn.lua b/lualib/lua_nn.lua index 0e8977d37..d0d2d5265 100644 --- a/lualib/lua_nn.lua +++ b/lualib/lua_nn.lua @@ -22,6 +22,7 @@ local lua_nn_models = {} if rspamd_config:has_torch() then torch = require "torch" + torch.setnumthreads(1) end if torch then |