gengdongjie
5 years ago
1 changed files with 0 additions and 6 deletions
Unified View
Diff Options
+ 0
- 6
mindspore/train/callback.py
View File
| @@ -686,9 +686,6 @@ class TimeMonitor(Callback): | |||||
| def __init__(self, data_size): | def __init__(self, data_size): | ||||
| super(TimeMonitor, self).__init__() | super(TimeMonitor, self).__init__() | ||||
| self.data_size = data_size | self.data_size = data_size | ||||
| self.step_time_cost = [] | |||||
| self.epoch_time_cost = [] | |||||
| self.per_step_time = [] | |||||
| def epoch_begin(self, run_context): | def epoch_begin(self, run_context): | ||||
| self.epoch_time = time.time() | self.epoch_time = time.time() | ||||
| @@ -696,8 +693,6 @@ class TimeMonitor(Callback): | |||||
| def epoch_end(self, run_context): | def epoch_end(self, run_context): | ||||
| epoch_mseconds = (time.time() - self.epoch_time) * 1000 | epoch_mseconds = (time.time() - self.epoch_time) * 1000 | ||||
| per_step_mseconds = epoch_mseconds / self.data_size | 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) | print("epoch time: {0}, per step time: {1}".format(epoch_mseconds, per_step_mseconds), flush=True) | ||||
| def step_begin(self, run_context): | def step_begin(self, run_context): | ||||
| @@ -705,6 +700,5 @@ class TimeMonitor(Callback): | |||||
| def step_end(self, run_context): | def step_end(self, run_context): | ||||
| step_mseconds = (time.time() - self.step_time) * 1000 | step_mseconds = (time.time() - self.step_time) * 1000 | ||||
| self.step_time_cost.append(step_mseconds) | |||||
| print('step time', step_mseconds, flush=True) | print('step time', step_mseconds, flush=True) | ||||