YuanGongND / vocalsound

Dataset and baseline code for the VocalSound dataset (ICASSP2022).
106 stars 10 forks source link

Bug in prep_data.py #3

Open jakekolessar opened 2 years ago

jakekolessar commented 2 years ago

When data_dir is used as an argument, it is never set as the data_dir variable

Suggested Update:

if __name__ == '__main__':
    args = parser.parse_args()
    # if no path is provided, use the default path ../data/
    if args.data_dir == None:
        cur_path = '/'.join(os.getcwd().split('/')[:-1])
        data_dir = cur_path + '/data'
    else:
        data_dir = args.data_dir
YuanGongND commented 2 years ago

Hi there,

Thanks so much for pointing this out.

I just fixed that according to your suggestion.

-Yuan