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 (