projects
/
dyncnn.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Use cudnn more efficiently.
[dyncnn.git]
/
dyncnn.lua
diff --git
a/dyncnn.lua
b/dyncnn.lua
index
839431a
..
e104386
100755
(executable)
--- a/
dyncnn.lua
+++ b/
dyncnn.lua
@@
-126,10
+126,10
@@
torch.manualSeed(opt.seed)
mynn = {}
mynn = {}
---
By default, mynn returns the entries from nn
+--
To deal elegantly with CPU/GPU
local mt = {}
function mt.__index(table, key)
local mt = {}
function mt.__index(table, key)
- return nn[key]
+ return
(cudnn and cudnn[key]) or (cunn and cunn[key]) or
nn[key]
end
setmetatable(mynn, mt)
end
setmetatable(mynn, mt)
@@
-144,8
+144,13
@@
if useGPU then
require 'cutorch'
require 'cunn'
require 'cudnn'
require 'cutorch'
require 'cunn'
require 'cudnn'
+
mynn.FastTensor = torch.CudaTensor
mynn.FastTensor = torch.CudaTensor
- mynn.SpatialConvolution = cudnn.SpatialConvolution
+
+ if cudnn then
+ cudnn.benchmark = true
+ cudnn.fastest = true
+ end
end
----------------------------------------------------------------------
end
----------------------------------------------------------------------