diff --git a/.gitignore b/.gitignore index 65916c7..683bdfc 100644 --- a/.gitignore +++ b/.gitignore @@ -21,3 +21,8 @@ rmvpe.pt # To set a Python version for the project .tool-versions + +/runtime +/assets/weights/* +ffmpeg.* +ffprobe.* \ No newline at end of file diff --git a/infer/modules/train/train.py b/infer/modules/train/train.py index ab6ba2d..c0ca0f6 100644 --- a/infer/modules/train/train.py +++ b/infer/modules/train/train.py @@ -104,10 +104,11 @@ def main(): os.environ["MASTER_ADDR"] = "localhost" os.environ["MASTER_PORT"] = str(randint(20000, 55555)) children = [] + logger = utils.get_logger(hps.model_dir) for i in range(n_gpus): subproc = mp.Process( target=run, - args=(i, n_gpus, hps), + args=(i, n_gpus, hps, logger), ) children.append(subproc) subproc.start() @@ -120,10 +121,11 @@ def run( rank, n_gpus, hps, + logger: logging.Logger ): global global_step if rank == 0: - logger = utils.get_logger(hps.model_dir) + # logger = utils.get_logger(hps.model_dir) logger.info(hps) # utils.check_git_hash(hps.model_dir) writer = SummaryWriter(log_dir=hps.model_dir)