Ejemplo n.º 1
0
def prepare_directories_and_logger(output_directory, log_directory, rank):
    if rank == 0:
        if not os.path.isdir(output_directory):
            os.makedirs(output_directory)
            os.chmod(output_directory, 0o775)
        logger = Tacotron2Logger(os.path.join(output_directory, log_directory))
    else:
        logger = None
    return logger
Ejemplo n.º 2
0
def prepare_directories_and_logger(hparams, args):
    if args.rank == 0:
        if not os.path.isdir(args.output_directory):
            os.makedirs(args.output_directory)
            os.chmod(args.output_directory, 0o775)
        logger = Tacotron2Logger(os.path.join(args.output_directory, args.log_directory), hparams)
    else:
        logger = None
    return logger
Ejemplo n.º 3
0
def prepare_directories_and_logger(output_directory, log_directory, rank):
    if rank == 0:
        # print("Is this getting executed?")
        if not os.path.isdir(output_directory):
            os.makedirs(output_directory)
            os.chmod(output_directory, 0o775)
        logger = Tacotron2Logger(log_directory)
    else:
        logger = None
    return logger
Ejemplo n.º 4
0
def prepare_directories_and_logger(haparms, output_directory, log_directory,
                                   rank):
    if rank == 0:
        if not os.path.isdir(output_directory):
            os.makedirs(output_directory)
            os.chmod(output_directory, 0o775)
        logger = Tacotron2Logger(os.path.join(output_directory, log_directory), \
            hparams.use_mmi, hparams.use_guided_attn_loss)
    else:
        logger = None
    return logger
Ejemplo n.º 5
0
def prepare_directories_and_logger(output_directory, log_directory, rank,
                                   run_name, prj_name, resume):
    if rank == 0:
        if not os.path.isdir(os.path.join(output_directory, prj_name,
                                          run_name)):
            os.makedirs(os.path.join(output_directory, prj_name, run_name))
            os.chmod(os.path.join(output_directory, prj_name, run_name), 0o775)
        logger = Tacotron2Logger(
            run_name, prj_name, os.path.join(log_directory, prj_name,
                                             run_name), resume)
    else:
        logger = None
    return logger
Ejemplo n.º 6
0
def prepare_directories_and_logger(output_directory, log_directory, rank):
    if rank == 0:
        log_path = os.path.join(output_directory, log_directory)
        try:
            shutil.rmtree(log_path)
        except:
            pass
        if not os.path.isdir(output_directory):
            os.makedirs(output_directory)
            os.chmod(output_directory, 0o777)
        logger = Tacotron2Logger(log_path)
    else:
        logger = None
    return logger
Ejemplo n.º 7
0
def prepare_directories_and_logger(output_directory, log_directory, rank,
                                   model_type):
    if rank == 0:
        if not os.path.isdir(output_directory):
            os.makedirs(output_directory)
            os.chmod(output_directory, 0o775)
        if model_type == "forwardtacotron":
            logger = ForwardTacotronLogger(
                os.path.join(output_directory, log_directory))
        else:
            logger = Tacotron2Logger(
                os.path.join(output_directory, log_directory))
    else:
        logger = None
    return logger
Ejemplo n.º 8
0
def prepare_directories_and_logger(output_directory, log_directory):
    if not os.path.isdir(output_directory):
        os.makedirs(output_directory)
        # os.chmod(output_directory, 0o775)
    logger = Tacotron2Logger(os.path.join(output_directory, log_directory))
    return logger