Update learning rates for all the networks; called at the end of every epoch
(self)
| 113 | return self.image_paths |
| 114 | |
| 115 | def update_learning_rate(self): |
| 116 | """Update learning rates for all the networks; called at the end of every epoch""" |
| 117 | for scheduler in self.schedulers: |
| 118 | if self.opt.lr_policy == 'plateau': |
| 119 | scheduler.step(self.metric) |
| 120 | else: |
| 121 | scheduler.step() |
| 122 | |
| 123 | #lr = self.optimizers[0].param_groups[0]['lr'] |
| 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""" |