| | |
|---|
| | | device = "cpu" |
|---|
| | | # kwargs["batch_size"] = 1 |
|---|
| | | kwargs["device"] = device |
|---|
| | | import pdb; pdb.set_trace() |
|---|
| | | |
|---|
| | | if kwargs.get("ncpu", None): |
|---|
| | | torch.set_num_threads(kwargs.get("ncpu")) |
|---|
| | |
|---|
| | | spk_embedding = result['spk_embedding'] |
|---|
| | | labels = self.cb_model(spk_embedding, oracle_num=self.preset_spk_num) |
|---|
| | | del result['spk_embedding'] |
|---|
| | | sv_output = postprocess(all_segments, None, labels, spk_embedding) |
|---|
| | | sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu()) |
|---|
| | | if self.spk_mode == 'vad_segment': |
|---|
| | | sentence_list = [] |
|---|
| | | for res, vadsegment in zip(restored_data, vadsegments): |
|---|
