瀏覽代碼

no confidence change for #128

nagadomi 9 年之前
父節點
當前提交
382d493514
共有 1 個文件被更改,包括 2 次插入3 次删除
  1. 2 3
      lib/w2nn.lua

+ 2 - 3
lib/w2nn.lua

@@ -7,18 +7,17 @@ local function load_cunn()
    require 'cunn'
 end
 local function load_cudnn()
-   require 'cudnn'
-   cudnn.benchmark = true
+   cudnn = require('cudnn')
 end
 if w2nn then
    return w2nn
 else
+   w2nn = {}
    local state, ret = pcall(load_cunn)
    if not state then
       error("Failed to load CUDA modules. Please check the CUDA Settings.\n---\n" .. ret)
    end
    pcall(load_cudnn)
-   w2nn = {}
 
    function w2nn.load_model(model_path, force_cudnn)
       local model = torch.load(model_path, "ascii")