1 files changed, 3 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index ca4d41d..06e57e1 100644
--- a/init.lua
+++ b/init.lua
@@ -688,5 +688,8 @@ for k,v in pairs(_G) do
    _G._preloaded_[k] = true
 end
 
+-- Enable heap tracking
+torch.setheaptracking(true)
+
 -- return repl, just call it to start it!
 return repl