工具代码

import argparse

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--sr", type=int, default=16000, help="sampling rate")
    parser.add_argument("--in_dir", type=str, default="vctk", help="path to source dir")
    args = parser.parse_args()

import os
from glob import glob
from tqdm import tqdm
from multiprocessing import Pool, cpu_count

def process(wav_name):
    # process wav_name
    pass

filenames = glob(f'{args.in_dir}/*/*.wav', recursive=True)
filenames = os.listdir(os.path.join(args.in_dir, speaker))

pool = Pool(processes=cpu_count()-2)

for _ in tqdm(pool.imap_unordered(process, filenames)):
    pass

def print_network(model, name):
    """Print out the network information."""
    num_params = 0
    for p in model.parameters():
        num_params += p.numel()
    print(model, flush=True)
    print(name,flush=True)
    print("The number of parameters: {}".format(num_params), flush=True)