浏览代码

Change default -validation_crops to 200.

nagadomi 9 年之前
父节点
当前提交
de669d24a4
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      train.lua

+ 3 - 2
train.lua

@@ -37,10 +37,11 @@ local function split_data(x, test_size)
 end
 local function make_validation_set(x, transformer, n, patches)
    n = n or 4
+   local validation_patches = math.min(16, patches or 16)
    local data = {}
    for i = 1, #x do
-      for k = 1, math.max(n / patches, 1) do
-	 local xy = transformer(x[i], true, patches)
+      for k = 1, math.max(n / validation_patches, 1) do
+	 local xy = transformer(x[i], true, validation_patches)
 	 for j = 1, #xy do
 	    table.insert(data, {x = xy[j][1], y = xy[j][2]})
 	 end