(self)
| 105 | self.optimizers.append(self.optimizer_F) |
| 106 | |
| 107 | def optimize_parameters(self): |
| 108 | # forward |
| 109 | self.forward() |
| 110 | |
| 111 | # update D |
| 112 | self.set_requires_grad(self.netD, True) |
| 113 | self.optimizer_D.zero_grad() |
| 114 | self.loss_D = self.compute_D_loss() |
| 115 | self.loss_D.backward() |
| 116 | self.optimizer_D.step() |
| 117 | |
| 118 | # update G |
| 119 | self.set_requires_grad(self.netD, False) |
| 120 | self.optimizer_G.zero_grad() |
| 121 | if self.opt.netF == 'mlp_sample': |
| 122 | self.optimizer_F.zero_grad() |
| 123 | self.loss_G = self.compute_G_loss() |
| 124 | self.loss_G.backward() |
| 125 | self.optimizer_G.step() |
| 126 | if self.opt.netF == 'mlp_sample': |
| 127 | self.optimizer_F.step() |
| 128 | |
| 129 | def set_input(self, input): |
| 130 | """Unpack input data from the dataloader and perform necessary pre-processing steps. |
nothing calls this directly
no test coverage detected