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

change time.time to time.time_ns

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