(self)
| 63 | betas=[self._opt.D_adam_b1, self._opt.D_adam_b2]) |
| 64 | |
| 65 | def _init_prefetch_inputs(self): |
| 66 | self._input_real_img = self._Tensor(self._opt.batch_size, 3, self._opt.image_size, self._opt.image_size) |
| 67 | self._input_real_cond = self._Tensor(self._opt.batch_size, self._opt.cond_nc) |
| 68 | self._input_desired_cond = self._Tensor(self._opt.batch_size, self._opt.cond_nc) |
| 69 | self._input_real_img_path = None |
| 70 | self._input_real_cond_path = None |
| 71 | |
| 72 | def _init_losses(self): |
| 73 | # define loss functions |