diff --git a/models/train/data_meeting_io.py b/models/train/data_meeting_io.py index 8bda1bb..060cfab 100644 --- a/models/train/data_meeting_io.py +++ b/models/train/data_meeting_io.py @@ -328,7 +328,7 @@ def utterance_info_generator(): (num_utterances, 1), (num_utterances, 1), (num_utterances, 1)] - tensor_type = [np.string_, np.int32, np.int32, np.float32, + tensor_type = [np.bytes_, np.int32, np.int32, np.float32, np.int32, np.float32] (wav_paths, diarizations, spkindices, meeting_scales, start_samples,