From 7e8e44ef56ee806f48f56685563576d1a5465525 Mon Sep 17 00:00:00 2001 From: candanzg Date: Fri, 24 Apr 2020 16:52:21 +0800 Subject: [PATCH] fixed summary graph bug Signed-off-by: candanzg --- mindspore/train/summary/summary_record.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/train/summary/summary_record.py b/mindspore/train/summary/summary_record.py index 3dbe31f0e4..4c60dce862 100644 --- a/mindspore/train/summary/summary_record.py +++ b/mindspore/train/summary/summary_record.py @@ -158,6 +158,9 @@ class SummaryRecord: package_graph_event(graph_proto).SerializeToString()) self.event_writer.flush() self.has_graph = True + data = _summary_tensor_cache.get("SummaryRecord") + if data is None: + return True data = _summary_tensor_cache.get("SummaryRecord") if data is None: