Enable heap tracking by default in torch

diz-vara · Jun 13, 2016 · d749e36 · d749e36
1 parent a167188
commit d749e36
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/init.lua b/init.lua
@@ -186,4 +186,7 @@ end
 -- alias for convenience
 torch.Tensor.isTensor = torch.isTensor
 
+-- remove this line to disable automatic heap-tracking for garbage collection
+torch.setheaptracking(true)
+
 return torch