barian0517/RCV
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化代码结构

Browse Source
This commit is contained in:
源文雨 2023-06-24 16:36:15 +08:00
parent 46c0e9b2fe
commit ccba65151e
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
extract_feature_print.py
View File

@ -19,14 +19,11 @@ import soundfile as sf
import numpy as np import numpy as np
from fairseq import checkpoint_utils from fairseq import checkpoint_utils
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = "cpu"
if torch.cuda.is_available(): if torch.cuda.is_available():
device = "cuda" device = "cuda"
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
device = "mps" device = "mps"
else:
device = "cpu"
f = open("%s/extract_f0_feature.log" % exp_dir, "a+") f = open("%s/extract_f0_feature.log" % exp_dir, "a+")