diff --git a/mindspore/train/summary/_event_writer.py b/mindspore/train/summary/_event_writer.py index ae347135f62a7e83dcced3c2dc5618915c5ff09b..0a21dea04ec2d693987fe9d6ff582852d3c20f93 100644 --- a/mindspore/train/summary/_event_writer.py +++ b/mindspore/train/summary/_event_writer.py @@ -47,7 +47,7 @@ class EventWriter(Process): def run(self): - with Pool() as pool: + with Pool(min(cpu_count(), 32)) as pool: deq = deque() while True: while deq and deq[0].ready(): diff --git a/mindspore/train/summary/summary_record.py b/mindspore/train/summary/summary_record.py index b2bc872a1fffa7e6c86edf3ceb83b9666472c4eb..2c9b687db472d54dfa72c6b1666e28b5c9896dad 100644 --- a/mindspore/train/summary/summary_record.py +++ b/mindspore/train/summary/summary_record.py @@ -89,6 +89,8 @@ class SummaryRecord: file_suffix="_MS", network=None): + self._event_writer, self._closed = None, False + _check_str_by_regular(file_prefix) _check_str_by_regular(file_suffix) self.log_path = _make_directory(log_dir) @@ -113,7 +115,6 @@ class SummaryRecord: self.suffix = file_suffix self.network = network self.has_graph = False - self._closed = False # create the summary writer file self.event_file_name = get_event_file_name(self.prefix, self.suffix) @@ -122,8 +123,6 @@ class SummaryRecord: except Exception as ex: raise RuntimeError(ex) - self._event_writer = None - def _init_event_writer(self): """Init event writer and write metadata.""" event_writer = EventWriter(self.full_file_name, self.flush_time)