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