fix workflow (#284)

* Update extract_feature_print.py

* Update unitest.yml
This commit is contained in:
源文雨 2023-05-15 13:11:01 +08:00 committed by GitHub
parent 137447bdc9
commit b07dedd744
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -32,5 +32,5 @@ jobs:
touch logs/mi-test/preprocess.log
python trainset_preprocess_pipeline_print.py logs/mute/0_gt_wavs 48000 8 logs/mi-test True
touch logs/mi-test/extract_f0_feature.log
python extract_f0_print.py logs/mi-test $(nproc) pm
python extract_f0_print.py logs/mi-test $(nproc) pm v1
python extract_feature_print.py cpu 1 0 0 logs/mi-test

View File

@ -5,11 +5,12 @@ n_part = int(sys.argv[2])
i_part = int(sys.argv[3])
if len(sys.argv) == 5:
exp_dir = sys.argv[4]
version = sys.argv[5]
else:
i_gpu = sys.argv[4]
exp_dir = sys.argv[5]
os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
version = sys.argv[6]
version = sys.argv[6]
import torch
import torch.nn.functional as F
import soundfile as sf