blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove the logging info on Terminal, redirect it

Browse Source 
to train.log
This commit is contained in:
XXXXRT666 2024-08-31 18:02:16 +08:00
parent 40b7f5197f
commit 3cdcc51adf
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/s2_train.py
View File

@ -306,7 +306,7 @@ def train_and_evaluate(
y_lengths,
text,
text_lengths,
) in enumerate(tqdm(train_loader,position=rank+1,leave=(epoch==hps.train.epochs),postfix=f'epoch:{epoch}',disable=(rank!=0))):
) in enumerate(tqdm(train_loader,position=rank,leave=(epoch==hps.train.epochs),postfix=f'Epoch:{epoch},Rank:{rank}',delay=1)):
if torch.cuda.is_available():
spec, spec_lengths = spec.cuda(rank, non_blocking=True), spec_lengths.cuda(
rank, non_blocking=True

6
GPT_SoVITS/utils.py
View File

@ -18,7 +18,7 @@ logging.getLogger("matplotlib").setLevel(logging.ERROR)
MATPLOTLIB_FLAG = False
logging.basicConfig(stream=sys.stdout, level=logging.ERROR)
# logging.basicConfig(stream=sys.stdout, level=logging.ERROR)
logger = logging
@ -319,13 +319,13 @@ def check_git_hash(model_dir):
def get_logger(model_dir, filename="train.log"):
global logger
logger = logging.getLogger(os.path.basename(model_dir))
logger.setLevel(logging.ERROR)
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter("%(asctime)s\t%(name)s\t%(levelname)s\t%(message)s")
if not os.path.exists(model_dir):
os.makedirs(model_dir)
h = logging.FileHandler(os.path.join(model_dir, filename))
h.setLevel(logging.ERROR)
h.setLevel(logging.DEBUG)
h.setFormatter(formatter)
logger.addHandler(h)
return logger