提交 16d14579 编写于 作者: S sneaxiy

change time.time to time.time_ns

上级 875c1f6d
......@@ -48,17 +48,17 @@ class ChunkTimer:
self.reset()
def reset(self):
self.begin = time.time()
self.begin = time.time_ns()
self.records.clear()
def start(self, name):
paddle.device.cuda.synchronize()
t = time.time()
t = time.time_ns()
self.records.append([name, t, None])
def end(self, name):
paddle.device.cuda.synchronize()
t = time.time()
t = time.time_ns()
self.records[-1][-1] = t
def export_info(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册