Update train_ms.py

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 (