Browse Source

!660 Fix bug: summary graph and record misjudgment

Merge pull request !660 from candanzg/summary_record_bug
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
5633f6c68b
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/train/summary/summary_record.py

+ 3
- 0
mindspore/train/summary/summary_record.py View File

@@ -158,6 +158,9 @@ class SummaryRecord:
package_graph_event(graph_proto).SerializeToString()) package_graph_event(graph_proto).SerializeToString())
self.event_writer.flush() self.event_writer.flush()
self.has_graph = True self.has_graph = True
data = _summary_tensor_cache.get("SummaryRecord")
if data is None:
return True


data = _summary_tensor_cache.get("SummaryRecord") data = _summary_tensor_cache.get("SummaryRecord")
if data is None: if data is None:


Loading…
Cancel
Save