From ae8243321160db570056f849ea68587acbc6096f Mon Sep 17 00:00:00 2001
From: farrajota <mglfarrajota@outlook.com>
Date: Mon, 10 Jul 2017 17:32:12 +0100
Subject: [PATCH] Set 'disable_cudnn' flag when loading netwrokds

---
 configs.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs.lua b/configs.lua
index 261c992..d507c9e 100644
--- a/configs.lua
+++ b/configs.lua
@@ -16,7 +16,7 @@ local function convert_model_backend(model, opt, is_gpu)
         model:cuda()
 
         -- require cudnn if available
-        if pcall(require, 'cudnn') then
+        if pcall(require, 'cudnn') and not opt.disable_cudnn then
             cudnn.convert(model, cudnn):cuda()
             cudnn.benchmark = true
             if opt.cudnn_deterministic then