diff options
| author | soumith <soumith@gmail.com> | 2015-09-17 19:42:39 +0300 |
|---|---|---|
| committer | soumith <soumith@gmail.com> | 2015-09-17 19:42:39 +0300 |
| commit | 4f480aeb2a34aa8833b7f89f0e3d9bdd9a2c0415 (patch) | |
| tree | 5b5eb3a889c57959bd823a09589e331c1bd71cf8 /init.lua | |
| parent | aab06d12e0ed361d696b268620796388ca8f59c7 (diff) | |
added getParameters to be on CPU
Diffstat (limited to 'init.lua')
| -rw-r--r-- | init.lua | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -5,3 +5,5 @@ require "libcunn" include('test.lua') include('DataParallelTable.lua') + +nn.Module._flattenTensorBuffer['torch.CudaTensor'] = torch.FloatTensor.new |