From c137d875cd07dd136587dde3d0c82327189f4e7f Mon Sep 17 00:00:00 2001
From: Sora <atri@suzakuintsubaki.com>
Date: Thu, 9 Nov 2023 17:38:15 +0800
Subject: [PATCH] use diy loguru (#164)

* Feat: use diy loguru logger

* Feat: use diy loguru logger
---
 data_utils.py | 2 +-
 tools/log.py  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/data_utils.py b/data_utils.py
index 6e85723..e94f69f 100644
--- a/data_utils.py
+++ b/data_utils.py
@@ -3,7 +3,7 @@ import random
 import torch
 import torch.utils.data
 from tqdm import tqdm
-from loguru import logger
+from tools.log import logger
 import commons
 from mel_processing import spectrogram_torch, mel_spectrogram_torch
 from utils import load_wav_to_torch, load_filepaths_and_text
diff --git a/tools/log.py b/tools/log.py
index 47e51ae..7fcccf2 100644
--- a/tools/log.py
+++ b/tools/log.py
@@ -10,7 +10,7 @@ logger.remove()
 
 # 自定义格式并添加到标准输出
 log_format = (
-    "<g>{time:MM-DD HH:mm:ss}</g> [<lvl>{level}</lvl>] | {file}:{line} | {message}"
+    "<g>{time:MM-DD HH:mm:ss}</g> <lvl>{level:<9}</lvl>| {file}:{line} | {message}"
 )
 
-logger.add(sys.stdout, format=log_format)
+logger.add(sys.stdout, format=log_format, backtrace=True, diagnose=True)