From 66bff68ef1d07dd0c20ba515fac96d3a0282845c Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Sat, 8 Oct 2016 17:22:16 +0900
Subject: [PATCH] split first

---
 train.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.lua b/train.lua
index bd173e3..22ac1c9 100644
--- a/train.lua
+++ b/train.lua
@@ -382,6 +382,8 @@ local function plot(train, valid)
 	 {'validation', torch.Tensor(valid), '-'}})
 end
 local function train()
+   local x = remove_small_image(torch.load(settings.images))
+   local train_x, valid_x = split_data(x, math.max(math.floor(settings.validation_rate * #x), 1))
    local hist_train = {}
    local hist_valid = {}
    local model
@@ -397,8 +399,6 @@ local function train()
 
    local criterion = create_criterion(model)
    local eval_metric = w2nn.ClippedMSECriterion(0, 1):cuda()
-   local x = remove_small_image(torch.load(settings.images))
-   local train_x, valid_x = split_data(x, math.max(math.floor(settings.validation_rate * #x), 1))
    local adam_config = {
       xLearningRate = settings.learning_rate,
       xBatchSize = settings.batch_size,