shixian.shi 2 jaren geleden
bovenliggende
commit
8cac1c97d9
1 gewijzigde bestanden met toevoegingen van 1 en 0 verwijderingen
  1. 1 0
      funasr/datasets/large_datasets/utils/padding.py

+ 1 - 0
funasr/datasets/large_datasets/utils/padding.py

@@ -14,6 +14,7 @@ def padding(data, float_pad_value=0.0, int_pad_value=-1):
     data_names = data[0].keys()
     for data_name in data_names:
         if data_name == "key" or data_name == "sampling_rate" or data_name == 'hotword_indxs':
+            batch[data_name] = data[0][data_name]
             continue
         else:
             if data[0][data_name].dtype.kind == "i":