diff --git a/egs/mars/sd/scripts/dump_rttm_to_labels.py b/egs/mars/sd/scripts/dump_rttm_to_labels.py index f254e8502..6b98d17bf 100644 --- a/egs/mars/sd/scripts/dump_rttm_to_labels.py +++ b/egs/mars/sd/scripts/dump_rttm_to_labels.py @@ -75,7 +75,7 @@ def calc_labels(spk_turns, spk_list, length, n_spk, remove_sil=False, max_overla sr=None, frame_shift=0.01): frame_shift = int(frame_shift * sr) num_frame = int((float(length) + (float(frame_shift) / 2)) / frame_shift) - multi_label = np.zeros([n_spk, num_frame], dtype=int) + multi_label = np.zeros([n_spk, num_frame], dtype=np.int32) for _, st, dur, spk in spk_turns: idx = spk_list.index(spk)