From e67ac621296cf6667d05b51f23ce8cb9ef8a0855 Mon Sep 17 00:00:00 2001 From: Lordmau5 Date: Sat, 15 Apr 2023 03:57:16 +0200 Subject: [PATCH] fix(train): suppress pytorch logs for deprecated TypedStorage (#336) --- src/so_vits_svc_fork/logger.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/so_vits_svc_fork/logger.py b/src/so_vits_svc_fork/logger.py index caa9110e..680c7524 100644 --- a/src/so_vits_svc_fork/logger.py +++ b/src/so_vits_svc_fork/logger.py @@ -1,5 +1,6 @@ import os import sys +import warnings from logging import ( DEBUG, INFO, @@ -35,6 +36,9 @@ def init_logger() -> None: if IS_TEST: getLogger(package_name).setLevel(DEBUG) captureWarnings(True) + warnings.filterwarnings( + "ignore", category=UserWarning, message="TypedStorage is deprecated" + ) LOGGER_INIT = True