parent
f6fa0c9cd9
commit
49434901d8
5
.gitignore
vendored
5
.gitignore
vendored
@ -21,3 +21,8 @@ rmvpe.pt
|
||||
|
||||
# To set a Python version for the project
|
||||
.tool-versions
|
||||
|
||||
/runtime
|
||||
/assets/weights/*
|
||||
ffmpeg.*
|
||||
ffprobe.*
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user