未验证 提交 b9b8c888 编写于 作者: W wanghuancoder 提交者: GitHub

use pre-commit formate code ptb_dy.py (#4871)

* fix ptb_dy time print for benchmark, test=develop

* use pre-commit formate code
上级 0739cc75
......@@ -37,20 +37,25 @@ if sys.version[0] == '2':
reload(sys)
sys.setdefaultencoding("utf-8")
class TimeCostAverage(object):
def __init__(self):
self.reset()
def reset(self):
self.cnt = 0
self.total_time = 0
def record(self, usetime):
self.cnt += 1
self.total_time += usetime
def get_average(self):
if self.cnt == 0:
return 0
return self.total_time / self.cnt
class SimpleLSTMRNN(fluid.Layer):
def __init__(self,
hidden_size,
......@@ -454,7 +459,8 @@ def train_ptb_lm():
"-- Epoch:[%d]; Batch:[%d]; ppl: %.5f, lr: %.5f, loss: %.5f, batch_cost: %.5f s, reader_cost: %.5f s"
% (epoch_id, batch_id, ppl[0],
sgd._global_learning_rate().numpy(), out_loss,
batch_cost_avg.get_average(), reader_cost_avg.get_average()))
batch_cost_avg.get_average(),
reader_cost_avg.get_average()))
batch_cost_avg.reset()
reader_cost_avg.reset()
batch_start = time.time()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册