def create_hparams(default_params, hparams_file=None, hparams_str=None):
hparams = HParams(**default_params)
if hparams_file is not None:
with open(hparams_file) as f:
hparams.parse_json(f.read())
if hparams_str is not None:
hparams.parse(hparams_str)
return hparams
def load_hparams(model_dir, default_params):
hparams_path = os.path.join(model_dir, HPARAMS_FILE)
hparams = HParams(default_params)
assert os.path.exists(hparams_path)
with open(hparams_path) as f:
return hparams.parse_json(f.read())
