Evaluate models using dataset.
custom_config = load_config(config_files)
// If the config file is empty, our config will be the base_config for the
// default model.
try:
model_type = custom_config["model"]["type"]
After Change
// If the config file is empty, our config will be the base_config for the
// default model.
try:
config = get_config(config_files, override_params=override_params)
except KeyError:
raise KeyError("model.type should be set on the custom config.")
config.train.job_dir = job_dir or config.train.job_dir