浏览代码

Reduce draw calls

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

+ 1 - 2
lib/minibatch_adam.lua

@@ -25,8 +25,6 @@ local function minibatch_adam(model, criterion, eval_metric,
       if t + batch_size -1 > train_x:size(1) then
 	 break
       end
-      xlua.progress(t, train_x:size(1))
-
       for i = 1, batch_size do
          inputs_tmp[i]:copy(train_x[shuffle[t + i - 1]])
 	 targets_tmp[i]:copy(train_y[shuffle[t + i - 1]])
@@ -50,6 +48,7 @@ local function minibatch_adam(model, criterion, eval_metric,
       c = c + 1
       if c % 50 == 0 then
 	 collectgarbage()
+	 xlua.progress(t, train_x:size(1))
       end
    end
    xlua.progress(train_x:size(1), train_x:size(1))