Return visualization images. train.py will display these images with visdom, and save the images to a HTML
(self)
| 124 | #print('learning rate = %.7f' % lr) |
| 125 | |
| 126 | def get_current_visuals(self): |
| 127 | """Return visualization images. train.py will display these images with visdom, and save the images to a HTML""" |
| 128 | visual_ret = OrderedDict() |
| 129 | for name in self.visual_names: |
| 130 | if isinstance(name, str): |
| 131 | visual_ret[name] = getattr(self, name) |
| 132 | return visual_ret |
| 133 | |
| 134 | def get_current_losses(self): |
| 135 | """Return traning losses / errors. train.py will print out these errors on console, and save them to a file""" |