diff options
| author | Soumith Chintala <soumith@gmail.com> | 2014-09-21 04:40:26 +0400 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2014-09-21 04:53:30 +0400 |
| commit | dcb0eee5c99ddb8264afa55f262ed23bb243281b (patch) | |
| tree | 61e43b8e51c35aa180178cd8ae7a35c821845d3d /test | |
| parent | 70433d6359cdae6833c315bb8151038ed9f75a1c (diff) | |
repro of #3
Diffstat (limited to 'test')
| -rw-r--r-- | test/clone.lua | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/test/clone.lua b/test/clone.lua new file mode 100644 index 0000000..3ea27cc --- /dev/null +++ b/test/clone.lua @@ -0,0 +1,13 @@ +require 'cudnn' + +i = torch.randn(128, 3, 100, 100):cuda() + +m1 = nn.Sequential() +m1:add(cudnn.SpatialConvolution(3,16,5,5)) +m1:add(cudnn.SpatialMaxPooling(2,2,2,2)) +m1:cuda() +m2 = m1:clone():cuda() + + +o1=m1:forward(i) +o2=m2:forward(i) |