Audio-AGI/AudioSep

Problem with the pipiline

Brodvd opened this issue · 0 comments

Hi @liuxubo717 I ranpipiline.py but where I find the checkpoint 3920000.ckpt? With the others checkpoint make error.

if __name__ == '__main__':
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    model = build_audiosep(
        config_yaml='config/audiosep_base.yaml', 
        checkpoint_path='checkpoint/step=3920000.ckpt', 
        device=device)