(self)
| 192 | self.val_ds.reset_state() |
| 193 | |
| 194 | def _trigger(self): |
| 195 | idx = 0 |
| 196 | for iA, iB in self.val_ds: |
| 197 | vizA, vizB = self.pred(iA, iB) |
| 198 | self.trainer.monitors.put_image('testA-{}'.format(idx), vizA) |
| 199 | self.trainer.monitors.put_image('testB-{}'.format(idx), vizB) |
| 200 | idx += 1 |
| 201 | |
| 202 | |
| 203 | if __name__ == '__main__': |