From bba911c8020e3272eb7986e62d3b42c4fd764c73 Mon Sep 17 00:00:00 2001 From: melgor Date: Wed, 16 Mar 2016 18:20:11 +0100 Subject: [PATCH] Replace satitize by clearState() --- training/{ => attic}/sanitize.lua | 0 training/train.lua | 3 +-- 2 files changed, 1 insertion(+), 2 deletions(-) rename training/{ => attic}/sanitize.lua (100%) 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