MemoryError

[LavenderMP]

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