(self, errors, step)
| 106 | |
| 107 | # errors: dictionary of error labels and values |
| 108 | def plot_current_errors(self, errors, step): |
| 109 | if self.tf_log: |
| 110 | for tag, value in errors.items(): |
| 111 | value = value.mean().float() |
| 112 | summary = self.tf.Summary(value=[self.tf.Summary.Value(tag=tag, simple_value=value)]) |
| 113 | self.writer.add_summary(summary, step) |
| 114 | |
| 115 | # errors: same format as |errors| of plotCurrentErrors |
| 116 | def print_current_errors(self, epoch, i, errors, t): |