(self)
| 25 | self.load_pretrained_models() |
| 26 | |
| 27 | def init_training_settings(self): |
| 28 | self.net_g.train() |
| 29 | train_opt = self.opt['train'] |
| 30 | |
| 31 | self.ema_decay = train_opt.get('ema_decay', 0) |
| 32 | if self.ema_decay > 0: |
| 33 | self.init_ema_model() |
| 34 | |
| 35 | self.load_pretrained_models() |
| 36 | self.setup_optimizers() |
| 37 | self.setup_schedulers() |
| 38 | self.setup_loss_functions() |
| 39 | |
| 40 | def load_pretrained_models(self): |
| 41 | load_path = self.opt['path'].get('pretrain_network_g', None) |
no test coverage detected