(self)
| 105 | return optimizer |
| 106 | |
| 107 | def test_start(self): |
| 108 | self.saving_result_pool = Pool(8) |
| 109 | self.saving_results_futures = [] |
| 110 | self.vocoder: BaseVocoder = get_vocoder_cls(hparams)() |
| 111 | if hparams.get('pe_enable') is not None and hparams['pe_enable']: |
| 112 | self.pe = PitchExtractor().cuda() |
| 113 | utils.load_ckpt(self.pe, hparams['pe_ckpt'], 'model', strict=True) |
| 114 | self.pe.eval() |
| 115 | def test_end(self, outputs): |
| 116 | self.saving_result_pool.close() |
| 117 | [f.get() for f in tqdm(self.saving_results_futures)] |
no test coverage detected