error when training 3-mix

[linshuijin123]

I try to train the model with 3-mix datasets, and I just change the num_speakers and data path, is there anything else I need to modify? thanks!

the following is the error detail:

Training...
data_loader.len 20000
/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/utils.py:45: UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than tensor.new_tensor(sourceTensor).
frame = signal.new_tensor(frame).long().cuda() # signal may in GPU or CPU
Traceback (most recent call last):
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/train.py", line 138, in
main(args)
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/train.py", line 132, in main
solver.train()
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/solver.py", line 60, in train
tr_avg_loss = self._run_one_epoch(epoch)
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/solver.py", line 153, in _run_one_epoch
cal_loss(padded_source, estimate_source, mixture_lengths)
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/pit_criterion.py", line 19, in cal_loss
source_lengths)
File "/home/bsipl_1/PycharmProjects/dual-path-RNNs-DPRNNs-based-speech-separation-master/pit_criterion.py", line 31, in cal_si_snr_with_pit
assert source.size() == estimate_source.size()
AssertionError

Process finished with exit code 1

[linshuijin123]

I get the same error when I run the other model with the same dataset