From d02814a057784cf997e281cad6f2a0a4f9eed27b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Sun, 20 Aug 2023 15:01:28 +0800
Subject: [PATCH] Update train_ms.py

---
 train_ms.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train_ms.py b/train_ms.py
index f8973c9..176c17d 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -13,7 +13,8 @@ import torch.distributed as dist
 from torch.nn.parallel import DistributedDataParallel as DDP
 from torch.cuda.amp import autocast, GradScaler
 from tqdm import tqdm
-
+import logging
+logging.getLogger('numba').setLevel(logging.WARNING)
 import commons
 import utils
 from data_utils import (