(self)
| 118 | |
| 119 | |
| 120 | def _load_model(self): |
| 121 | self.model = build_network(self.opt['network_g'], False).to(self.device) |
| 122 | self.model.eval() |
| 123 | self.model = self.model.to(self.device) |
| 124 | load_path = self.opt['path'].get('pretrain_network_g') |
| 125 | if load_path: |
| 126 | self.load_network(self.model, load_path, strict=False) |
| 127 | |
| 128 | def _register_hooks(self): |
| 129 | module_name_list = [] |
no test coverage detected