(self, outputs)
| 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)] |
| 118 | self.saving_result_pool.join() |
| 119 | return {} |
| 120 | |
| 121 | ########## |
| 122 | # utils |