|
|
|
@@ -686,9 +686,6 @@ class TimeMonitor(Callback): |
|
|
|
def __init__(self, data_size): |
|
|
|
super(TimeMonitor, self).__init__() |
|
|
|
self.data_size = data_size |
|
|
|
self.step_time_cost = [] |
|
|
|
self.epoch_time_cost = [] |
|
|
|
self.per_step_time = [] |
|
|
|
|
|
|
|
def epoch_begin(self, run_context): |
|
|
|
self.epoch_time = time.time() |
|
|
|
@@ -696,8 +693,6 @@ class TimeMonitor(Callback): |
|
|
|
def epoch_end(self, run_context): |
|
|
|
epoch_mseconds = (time.time() - self.epoch_time) * 1000 |
|
|
|
per_step_mseconds = epoch_mseconds / self.data_size |
|
|
|
self.epoch_time_cost.append(epoch_mseconds) |
|
|
|
self.per_step_time.append(per_step_mseconds) |
|
|
|
print("epoch time: {0}, per step time: {1}".format(epoch_mseconds, per_step_mseconds), flush=True) |
|
|
|
|
|
|
|
def step_begin(self, run_context): |
|
|
|
@@ -705,6 +700,5 @@ class TimeMonitor(Callback): |
|
|
|
|
|
|
|
def step_end(self, run_context): |
|
|
|
step_mseconds = (time.time() - self.step_time) * 1000 |
|
|
|
self.step_time_cost.append(step_mseconds) |
|
|
|
print('step time', step_mseconds, flush=True) |
|
|
|
|