This commit is contained in:
ylzz1997 2023-07-22 23:02:52 +08:00
parent 4961bf9657
commit befc4593c3
2 changed files with 3 additions and 2 deletions

View File

@ -5,12 +5,12 @@ import random
from concurrent.futures import ProcessPoolExecutor
from glob import glob
from random import shuffle
from loguru import logger
import librosa
import numpy as np
import torch
import torch.multiprocessing as mp
from loguru import logger
from tqdm import tqdm
import diffusion.logger.utils as du

View File

@ -1,6 +1,7 @@
import argparse
import torch
from loguru import logger
from torch.optim import lr_scheduler
from diffusion.data_loaders import get_data_loaders
@ -8,7 +9,7 @@ from diffusion.logger import utils
from diffusion.solver import train
from diffusion.unit2mel import Unit2Mel
from diffusion.vocoder import Vocoder
from loguru import logger
def parse_args(args=None, namespace=None):
"""Parse command-line arguments."""