tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

适配windows

Browse Source
This commit is contained in:
Stardust·减
2023-09-03 00:42:26 +08:00
committed by GitHub
parent db74ac71c2
commit d81876c456
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bert_gen.py
View File

@@ -5,7 +5,7 @@ import commons
import utils import utils
from data_utils import TextAudioSpeakerLoader, TextAudioSpeakerCollate from data_utils import TextAudioSpeakerLoader, TextAudioSpeakerCollate
from tqdm import tqdm from tqdm import tqdm
from multiprocessing import Pool import warnings
from text import cleaned_text_to_sequence, get_bert from text import cleaned_text_to_sequence, get_bert
@@ -40,6 +40,7 @@ def process_line(line):
torch.save(bert, bert_path) torch.save(bert, bert_path)
if __name__ == '__main__':
lines = [] lines = []
with open(hps.data.training_files) as f: with open(hps.data.training_files) as f:
lines.extend(f.readlines()) lines.extend(f.readlines())
@@ -47,7 +48,6 @@ with open(hps.data.training_files) as f:
with open(hps.data.validation_files) as f: with open(hps.data.validation_files) as f:
lines.extend(f.readlines()) lines.extend(f.readlines())
with Pool(processes=12) as pool: #A100 suitable config,if coom,please decrease the processess number. with Pool(processes=12) as pool: #A100 suitable config,if coom,please decrease the processess number.
for _ in tqdm(pool.imap_unordered(process_line, lines)): for _ in tqdm(pool.imap_unordered(process_line, lines)):
pass pass