提交 7c7b0c4d 编写于 作者: D dengkaipeng

fix parallel.

上级 243055bb
...@@ -110,7 +110,7 @@ def train(): ...@@ -110,7 +110,7 @@ def train():
for iter_id in range(cfg.start_iter, cfg.max_iter): for iter_id in range(cfg.start_iter, cfg.max_iter):
prev_start_time = start_time prev_start_time = start_time
start_time = time.time() start_time = time.time()
losses = exe.run(fetch_list=[v.name for v in fetch_list]) losses = exe.run(compile_program, fetch_list=[v.name for v in fetch_list])
smoothed_loss.add_value(np.mean(np.array(losses[0]))) smoothed_loss.add_value(np.mean(np.array(losses[0])))
snapshot_loss += np.mean(np.array(losses[0])) snapshot_loss += np.mean(np.array(losses[0]))
snapshot_time += start_time - prev_start_time snapshot_time += start_time - prev_start_time
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册