[重大bug]修复单说话人数据集加载emotion问题

innnky · Mar 30, 2023 · 09e1654 · 09e1654
1 parent 0bdc2ef
commit 09e1654
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_utils.py b/data_utils.py
@@ -148,7 +148,7 @@ def __call__(self, batch):
             wav_padded[i, :, :wav.size(1)] = wav
             wav_lengths[i] = wav.size(1)
 
-            emo[:] = row[3]
+            emo[i, :] = row[3]
 
         if self.return_ids:
             return text_padded, text_lengths, spec_padded, spec_lengths, wav_padded, wav_lengths, ids_sorted_decreasing