From d8feaa9c19e7acb89c17186f90ac202756d25f1f Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Sat, 18 Apr 2020 11:44:08 +0800 Subject: [PATCH] remove list append to avoid memory explosion --- mindspore/train/callback.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/mindspore/train/callback.py b/mindspore/train/callback.py index dcf630342c..b9635acc62 100644 --- a/mindspore/train/callback.py +++ b/mindspore/train/callback.py @@ -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)