未验证 提交 d6956cab 编写于 作者: L leaves-zwx 提交者: GitHub

Merge pull request #222 from Oneflow-Inc/dev_fix_gpt_kv_store

Zero after output
......@@ -174,6 +174,8 @@ class Metric(object):
self.print_fn_()
for key in self.keys_:
self.kv_store_[key] = 0.0
self.acc_elapsed_time_ = 0.0
self.acc_micro_batches_ = 0
self.acc_samples_ = 0
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册