nagadomi пре 9 година
родитељ
комит
fa9355be7c
1 измењених фајлова са 6 додато и 0 уклоњено
  1. 6 0
      train.lua

+ 6 - 0
train.lua

@@ -48,6 +48,12 @@ local function make_validation_set(x, transformer, n, patches)
       xlua.progress(i, #x)
       collectgarbage()
    end
+   local new_data = {}
+   local perm = torch.randperm(#data)
+   for i = 1, perm:size(1) do
+      new_data[i] = data[perm[i]]
+   end
+   data = new_data
    return data
 end
 local function validate(model, criterion, data, batch_size)