Преглед изворни кода

Use similar distribution on train/validation

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

+ 2 - 2
train.lua

@@ -95,8 +95,8 @@ local function transformer(x, is_validation, n, offset)
    local active_cropping_rate = nil
    local active_cropping_tries = nil
    if is_validation then
-      active_cropping_rate = 0
-      active_cropping_tries = 0
+      active_cropping_rate = settings.active_cropping_rate
+      active_cropping_tries = settings.active_cropping_tries
       random_color_noise_rate = 0.0
       random_overlay_rate = 0.0
    else