Spaces:
Running
Running
| import os | |
| import torch | |
| import argparse | |
| import subprocess | |
| assert torch.cuda.is_available(), "\033[31m You need GPU to Train! \033[0m" | |
| print("CPU Count is :", os.cpu_count()) | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("-t", type=int, default=0, help="thread count") | |
| args = parser.parse_args() | |
| commands = [ | |
| "python prepare/preprocess_a.py -w ./dataset_raw -o ./data_svc/waves-16k -s 16000 -t 0", | |
| "python prepare/preprocess_a.py -w ./dataset_raw -o ./data_svc/waves-32k -s 32000 -t 0", | |
| "python prepare/preprocess_crepe.py -w data_svc/waves-16k/ -p data_svc/pitch", | |
| "python prepare/preprocess_ppg.py -w data_svc/waves-16k/ -p data_svc/whisper", | |
| "python prepare/preprocess_hubert.py -w data_svc/waves-16k/ -v data_svc/hubert", | |
| "python prepare/preprocess_speaker.py data_svc/waves-16k/ data_svc/speaker -t 0", | |
| "python prepare/preprocess_speaker_ave.py data_svc/speaker/ data_svc/singer", | |
| "python prepare/preprocess_spec.py -w data_svc/waves-32k/ -s data_svc/specs -t 0", | |
| "python prepare/preprocess_train.py", | |
| "python prepare/preprocess_zzz.py", | |
| ] | |
| for command in commands: | |
| print(f"Command: {command}") | |
| process = subprocess.Popen(command, shell=True) | |
| outcode = process.wait() | |
| if (outcode): | |
| break | |