• Robin Scheibler's avatar
    Fixes examples/source_separation for WSJ0_2mix dataset (#2987) · f9d38796
    Robin Scheibler authored
    Summary:
    The `examples/source_separation` scripts use inconsistent keyword to indicate the WSJ0_2mix dataset. This PR does the following.
    
    1. Use `wsj0mix` consistently as keyword indicating the WSJ0_2mix dataset
    2. Corrects `args.data_dir` to `args.root_dir` in eval.py
    3. Modify the parameters of `pytorch_lightning.Trainer` according to latest version (use `accelerator="gpu"` and `devices=args.num_devices`, instead of just `gpus=args.num_devices`)
    
    Pull Request resolved: https://github.com/pytorch/audio/pull/2987
    
    Reviewed By: xiaohui-zhang
    
    Differential Revision: D42536992
    
    Pulled By: nateanl
    
    fbshipit-source-id: 10a80263ad7054b1629d8fa023676b607e633d76
    f9d38796
eval.py 3.28 KB