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
2 changed files with 3 additions and 2 deletions

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