From c52e0804032bddb55a1a0259fcf268f9df6407ce Mon Sep 17 00:00:00 2001 From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com> Date: Thu, 2 May 2024 02:04:19 +0100 Subject: [PATCH] Update utils.py now we can control the proportion --- GPT_SoVITS/AR/models/utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/GPT_SoVITS/AR/models/utils.py b/GPT_SoVITS/AR/models/utils.py index f4b39b9c..cd27552e 100644 --- a/GPT_SoVITS/AR/models/utils.py +++ b/GPT_SoVITS/AR/models/utils.py @@ -208,17 +208,17 @@ def make_reject_y(y_o, y_lens): bs = len(y_lens) reject_y = [] reject_y_lens = [] - process_item_idx = True + proportion=0.5 # repeat/total for b in range(bs): - if process_item_idx == True: + process_item_idx = torch.rand(1) + if process_item_idx < proportion: new_y = repeat_P(y_o[b]) reject_y.append(new_y) reject_y_lens.append(len(new_y)) - elif process_item_idx == False: + else: new_y = lost_P(y_o[b]) reject_y.append(new_y) reject_y_lens.append(len(new_y)) - process_item_idx = not process_item_idx max_length = max(reject_y_lens) for b in range(bs): pad_length = max_length - reject_y_lens[b]