Пример #1
0
def parse_model_args(model_name, parser, add_help=False):
    if model_name == 'Tacotron2':
        return parse_tacotron2_args(parser, add_help)
    if model_name == 'WaveGlow':
        return parse_waveglow_args(parser, add_help)
    else:
        raise NotImplementedError(model_name)
Пример #2
0
def parse_model_args(model_name, parser, add_help=False):
    if model_name == 'Tacotron2':
        from tacotron2.arg_parser import parse_tacotron2_args
        return parse_tacotron2_args(parser, add_help)
    if model_name == 'WaveGlow':
        from waveglow.arg_parser import parse_waveglow_args
        return parse_waveglow_args(parser, add_help)
    elif model_name == 'FastPitch':
        from fastpitch.arg_parser import parse_fastpitch_args
        return parse_fastpitch_args(parser, add_help)
    else:
        raise NotImplementedError(model_name)