提交 f97adda3 编写于 作者: L liuhongyu

fix gru lm

上级 8ea42994
...@@ -148,7 +148,7 @@ def train(train_reader, ...@@ -148,7 +148,7 @@ def train(train_reader,
if pass_idx == pass_num - 1 and args.enable_ce: if pass_idx == pass_num - 1 and args.enable_ce:
#Note: The following logs are special for CE monitoring. #Note: The following logs are special for CE monitoring.
#Other situations do not need to care about these logs. #Other situations do not need to care about these logs.
gpu_num = get_cards(args.enable_ce) gpu_num = get_cards(args)
if gpu_num == 1: if gpu_num == 1:
print("kpis imikolov_20_pass_duration %s" % print("kpis imikolov_20_pass_duration %s" %
(total_time / epoch_idx)) (total_time / epoch_idx))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册