diff --git a/infer-web.py b/infer-web.py index 9cfa723..b75b8f8 100644 --- a/infer-web.py +++ b/infer-web.py @@ -711,11 +711,31 @@ def train1key( ####### step2a:提取音高 yield get_info_str(i18n("step2:正在提取音高&正在提取特征")) - [get_info_str(_) for _ in extract_f0_feature(gpus16, np7, f0method8, if_f0_3, exp_dir1, version19, gpus_rmvpe)] + [ + get_info_str(_) + for _ in extract_f0_feature( + gpus16, np7, f0method8, if_f0_3, exp_dir1, version19, gpus_rmvpe + ) + ] ####### step3a:训练模型 yield get_info_str(i18n("step3a:正在训练模型")) - click_train(exp_dir1, sr2, if_f0_3, spk_id5, save_epoch10, total_epoch11, batch_size12, if_save_latest13, pretrained_G14, pretrained_D15, gpus16, if_cache_gpu17, if_save_every_weights18, version19) + click_train( + exp_dir1, + sr2, + if_f0_3, + spk_id5, + save_epoch10, + total_epoch11, + batch_size12, + if_save_latest13, + pretrained_G14, + pretrained_D15, + gpus16, + if_cache_gpu17, + if_save_every_weights18, + version19, + ) yield get_info_str(i18n("训练结束, 您可查看控制台训练日志或实验文件夹下的train.log")) ####### step3b:训练索引 diff --git a/infer/modules/train/extract/extract_f0_print.py b/infer/modules/train/extract/extract_f0_print.py index a3e9c9b..6949f1c 100644 --- a/infer/modules/train/extract/extract_f0_print.py +++ b/infer/modules/train/extract/extract_f0_print.py @@ -10,6 +10,7 @@ import logging import numpy as np import pyworld + from infer.lib.audio import load_audio logging.getLogger("numba").setLevel(logging.WARNING) diff --git a/infer/modules/train/extract/extract_f0_rmvpe.py b/infer/modules/train/extract/extract_f0_rmvpe.py index 98a91e7..52d7492 100644 --- a/infer/modules/train/extract/extract_f0_rmvpe.py +++ b/infer/modules/train/extract/extract_f0_rmvpe.py @@ -10,6 +10,7 @@ import logging import numpy as np import pyworld + from infer.lib.audio import load_audio logging.getLogger("numba").setLevel(logging.WARNING) diff --git a/infer/modules/train/extract/extract_f0_rmvpe_dml.py b/infer/modules/train/extract/extract_f0_rmvpe_dml.py index 2d5c7f4..2d812ab 100644 --- a/infer/modules/train/extract/extract_f0_rmvpe_dml.py +++ b/infer/modules/train/extract/extract_f0_rmvpe_dml.py @@ -10,6 +10,7 @@ import logging import numpy as np import pyworld + from infer.lib.audio import load_audio logging.getLogger("numba").setLevel(logging.WARNING) diff --git a/infer/modules/train/preprocess.py b/infer/modules/train/preprocess.py index 7ff76ee..c57b5dc 100644 --- a/infer/modules/train/preprocess.py +++ b/infer/modules/train/preprocess.py @@ -18,9 +18,10 @@ import traceback import librosa import numpy as np +from scipy.io import wavfile + from infer.lib.audio import load_audio from infer.lib.slicer2 import Slicer -from scipy.io import wavfile mutex = multiprocessing.Lock() f = open("%s/preprocess.log" % exp_dir, "a+")