search

taylorlu / Speaker-Diarization

speaker diarization by uis-rnn and speaker embedding by vgg-speaker-recognition
Apache License 2.0
455 stars 124 forks source link

MemoryError #22

Closed LavenderMP closed 4 years ago

LavenderMP commented 4 years ago

Hi taylorlu could you help me out of this problem. I did follow your step to train on custom dataset. the code itself works well on VCTK dataset (for testing) and then i did same step to apply on my custom dataset and it show this error: Traceback (most recent call last): File "train.py", line 91, in main() File "train.py", line 87, in main diarization_experiment(model_args, training_args, inference_args) File "train.py", line 48, in diarization_experiment model.fit(train_sequence_list, train_cluster_id_list, training_args) File "/home/huynh_hoang/Speaker-Diarization/uisrnn/uisrnn.py", line 373, in fit concatenated_train_sequence, concatenated_train_cluster_id, args) File "/home/huynh_hoang/Speaker-Diarization/uisrnn/uisrnn.py", line 240, in fit_concatenated num_permutations=args.num_permutations) File "/home/huynh_hoang/Speaker-Diarization/uisrnn/utils.py", line 214, in resize_sequence sub_sequences.append(sequence[sampled_idx_sets[j], :]) MemoryError: Unable to allocate array with shape (27848, 512) and data type float64