fixed summary graph bug

Signed-off-by: candanzg <zhangshucheng@huawei.com>
5 years ago · 7e8e44ef56
parent 96d39886b9
commit 7e8e44ef56
1 changed files with 3 additions and 0 deletions
--- 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: