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 = (
- "{time:MM-DD HH:mm:ss} [{level}] | {file}:{line} | {message}"
+ "{time:MM-DD HH:mm:ss} {level:<9}| {file}:{line} | {message}"
)
-logger.add(sys.stdout, format=log_format)
+logger.add(sys.stdout, format=log_format, backtrace=True, diagnose=True)