About the training code

[wenyuc55]

Hi, when I was training the code, I couldn't separate all ten audios. Only the first audio could be separated, but the second one couldn't be separated. If I change the code to start separation from the second audio, the third audio cannot be separated. This means that only the first one can be separated each time, and it will be forced to stop when the next one is executed. I executed it in Ubuntu to confirm that it was not a problem of insufficient memory.