File size: 1,426 Bytes
ad16788 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# -*- coding: utf-8 -*-
# Copyright 2018 Nagoya University (Tomoki Hayashi)
# Apache 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
import argparse
import logging
def fill_missing_args(args, add_arguments):
"""Fill missing arguments in args.
Args:
args (Namespace or None): Namesapce containing hyperparameters.
add_arguments (function): Function to add arguments.
Returns:
Namespace: Arguments whose missing ones are filled with default value.
Examples:
>>> from argparse import Namespace
>>> from espnet.nets.pytorch_backend.e2e_tts_tacotron2 import Tacotron2
>>> args = Namespace()
>>> fill_missing_args(args, Tacotron2.add_arguments_fn)
Namespace(aconv_chans=32, aconv_filts=15, adim=512, atype='location', ...)
"""
# check argument type
assert isinstance(args, argparse.Namespace) or args is None
assert callable(add_arguments)
# get default arguments
default_args, _ = add_arguments(argparse.ArgumentParser()).parse_known_args()
# convert to dict
args = {} if args is None else vars(args)
default_args = vars(default_args)
for key, value in default_args.items():
if key not in args:
logging.info(
'attribute "%s" does not exist. use default %s.' % (key, str(value))
)
args[key] = value
return argparse.Namespace(**args)
|