fix(train): suppress pytorch logs for deprecated TypedStorage (#336)

voicepaw · Apr 15, 2023 · e67ac62 · e67ac62
1 parent c9025ed
commit e67ac62
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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