Set 'disable_cudnn' flag when loading netwrokds

farrajota · Jul 10, 2017 · ae82433 · ae82433
1 parent 4e7acd8
commit ae82433
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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