diff --git a/training/sanitize.lua b/training/attic/sanitize.lua
similarity index 100%
rename from training/sanitize.lua
rename to training/attic/sanitize.lua
diff --git a/training/train.lua b/training/train.lua
index cd788c0..6b33bef 100644
--- a/training/train.lua
+++ b/training/train.lua
@@ -22,7 +22,6 @@ require 'torchx' --for concetration the table of tensors
 
 paths.dofile("OpenFaceOptim.lua")
 
-local sanitize = paths.dofile('sanitize.lua')
 
 local optimMethod = optim.adadelta
 local optimState = {} -- Use for other algorithms like SGD
@@ -89,7 +88,7 @@ function train()
 
    collectgarbage()
    
-   local nnModel = sanitize(model:float():clone())
+   local nnModel = model:float():clone():clearState()
    if opt.cudnn then
     cudnn.convert(nnModel,nn)
    end