diff options
| author | Adam Lerer <alerer@fb.com> | 2015-08-04 23:42:34 +0300 |
|---|---|---|
| committer | Adam Lerer <alerer@fb.com> | 2015-08-19 21:04:56 +0300 |
| commit | 843374d9587c2493268b3a0e73fa9c9819b5a3a5 (patch) | |
| tree | 255580888a56f5de12c9ddbc13153a4b78c483ed /init.lua | |
| parent | a16af4be2fee0e987bff6a0525e869fcfdb9b03d (diff) | |
cutorch gc
Diffstat (limited to 'init.lua')
| -rw-r--r-- | init.lua | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -29,4 +29,6 @@ function cutorch.createCudaHostTensor(...) return torch.FloatTensor(storage, 1, size:storage()) end +cutorch.setHeapTracking(true) + return cutorch |