diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-08-25 02:32:43 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-08-25 02:32:43 +0300 |
commit | 07b0b2d9989c7f68dad2b9d43eb0813315c7b032 (patch) | |
tree | 1fc2d75e2f1381a9d2cd89dc47713db7f200b35a /convert.lua | |
parent | 9c782eaa018335a696876fd368965852a3a265cd (diff) |
add error for nngraph
Diffstat (limited to 'convert.lua')
-rw-r--r-- | convert.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/convert.lua b/convert.lua index 49a6fa5..0a5b397 100644 --- a/convert.lua +++ b/convert.lua @@ -22,6 +22,9 @@ local layer_list = { -- for example: net = cudnn.convert(net, cudnn) function cudnn.convert(net, dst, exclusion_fn) return net:replace(function(x) + if torch.type(x) == 'nn.gModule' then + error('cudnn.convert does not work with nngraph yet') + end local y = 0 local src = dst == nn and cudnn or nn local src_prefix = src == nn and 'nn.' or 'cudnn.' |