tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-20 15:01:28 +08:00
committed by GitHub
parent 323eb0298d
commit d02814a057
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_ms.py
View File

@@ -13,7 +13,8 @@ import torch.distributed as dist
from torch.nn.parallel import DistributedDataParallel as DDP from torch.nn.parallel import DistributedDataParallel as DDP
from torch.cuda.amp import autocast, GradScaler from torch.cuda.amp import autocast, GradScaler
from tqdm import tqdm from tqdm import tqdm
import logging
logging.getLogger('numba').setLevel(logging.WARNING)
import commons import commons
import utils import utils
from data_utils import ( from data_utils import (