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)
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)