(self)
| 370 | self._save_optimizer(self._optimizer_D, 'D', label) |
| 371 | |
| 372 | def load(self): |
| 373 | load_epoch = self._opt.load_epoch |
| 374 | |
| 375 | # load G |
| 376 | self._load_network(self._G, 'G', load_epoch) |
| 377 | |
| 378 | if self._is_train: |
| 379 | # load D |
| 380 | self._load_network(self._D, 'D', load_epoch) |
| 381 | |
| 382 | # load optimizers |
| 383 | self._load_optimizer(self._optimizer_G, 'G', load_epoch) |
| 384 | self._load_optimizer(self._optimizer_D, 'D', load_epoch) |
| 385 | |
| 386 | def update_learning_rate(self): |
| 387 | # updated learning rate G |
no test coverage detected