From 066068f9a4d4f6b4b4fb2e0037df84451731e061 Mon Sep 17 00:00:00 2001 From: jiangshuqiang Date: Wed, 14 Apr 2021 11:47:18 +0800 Subject: [PATCH] fix queue.empty block for summary --- mindspore/train/summary/_writer_pool.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/mindspore/train/summary/_writer_pool.py b/mindspore/train/summary/_writer_pool.py index ec11b86351..2ea7bbff25 100644 --- a/mindspore/train/summary/_writer_pool.py +++ b/mindspore/train/summary/_writer_pool.py @@ -16,6 +16,7 @@ import os import time import signal +import queue from collections import deque import psutil @@ -108,14 +109,17 @@ class WriterPool(ctx.Process): for plugin, data in deq.popleft().get(): self._write(plugin, data) - if not self._queue.empty(): - action, data = self._queue.get() + try: + action, data = self._queue.get(block=False) if action == 'WRITE': deq.append(pool.apply_async(_pack_data, (data, time.time()))) elif action == 'FLUSH': self._flush() elif action == 'END': break + except queue.Empty: + pass + for result in deq: for plugin, data in result.get(): self._write(plugin, data)