| 64 | |
| 65 | |
| 66 | class MergeAllSummaries_RunAlone(Callback): |
| 67 | def __init__(self, period, key): |
| 68 | self._period = period |
| 69 | self._key = key |
| 70 | |
| 71 | def _setup_graph(self): |
| 72 | size = len(tf.get_collection(self._key)) |
| 73 | logger.info("Summarizing collection '{}' of size {}.".format(self._key, size)) |
| 74 | self.summary_op = tf.summary.merge_all(self._key) |
| 75 | |
| 76 | def _trigger_step(self): |
| 77 | if self._period: |
| 78 | if (self.local_step + 1) % self._period == 0: |
| 79 | self._trigger() |
| 80 | |
| 81 | def _trigger(self): |
| 82 | if self.summary_op: |
| 83 | summary = self.summary_op.eval() |
| 84 | self.trainer.monitors.put_summary(summary) |
| 85 | |
| 86 | |
| 87 | class MergeAllSummaries_RunWithOp(Callback): |