bytedance / music_source_separation

Other
1.24k stars 192 forks source link

Save the checkpoint with best testing SDR #36

Open sophia1488 opened 2 years ago

sophia1488 commented 2 years ago

The original code saved checkpoint every save_step_frequency steps, and I think it'd be helpful if the model is saved according to best testing SDR. Note that this PR assumes there's only one target source, like accompaniment or vocals.