(self, epoch, current_iter)
| 170 | |
| 171 | imwrite(sr_img, save_img_path) |
| 172 | def save(self, epoch, current_iter): |
| 173 | if hasattr(self, 'net_g_ema'): |
| 174 | self.save_network([self.net_g, self.net_g_ema], 'net_g', current_iter, param_key=['params', 'params_ema']) |
| 175 | else: |
| 176 | self.save_network(self.net_g, 'net_g', current_iter) |
| 177 | self.save_training_state(epoch, current_iter) |
nothing calls this directly
no test coverage detected