소스 검색

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
 end
 local function make_validation_set(x, transformer, n, patches)
 local function make_validation_set(x, transformer, n, patches)
    n = n or 4
    n = n or 4
+   local validation_patches = math.min(16, patches or 16)
    local data = {}
    local data = {}
    for i = 1, #x do
    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
 	 for j = 1, #xy do
 	    table.insert(data, {x = xy[j][1], y = xy[j][2]})
 	    table.insert(data, {x = xy[j][1], y = xy[j][2]})
 	 end
 	 end