(self, epoch, i, errors, t)
| 114 | |
| 115 | # errors: same format as |errors| of plotCurrentErrors |
| 116 | def print_current_errors(self, epoch, i, errors, t): |
| 117 | message = '(epoch: %d, iters: %d, time: %.3f) ' % (epoch, i, t) |
| 118 | for k, v in errors.items(): |
| 119 | #print(v) |
| 120 | #if v != 0: |
| 121 | v = v.mean().float() |
| 122 | message += '%s: %.3f ' % (k, v) |
| 123 | |
| 124 | print(message) |
| 125 | with open(self.log_name, "a") as log_file: |
| 126 | log_file.write('%s\n' % message) |
| 127 | |
| 128 | def convert_visuals_to_numpy(self, visuals): |
| 129 | for key, t in visuals.items(): |