From 609d4f862202c43241fdab06b0795db6281ca3f7 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Sun, 17 Sep 2017 10:04:59 +0100
Subject: [Feature] Allow to specify number of threads for ANN learning

---
 lualib/lua_nn.lua | 1 +
 1 file changed, 1 insertion(+)

(limited to 'lualib')

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
-- 
cgit v1.2.3