Skip to content
Snippets Groups Projects
Commit a95b53f8 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI Committed by Tiago de Freitas Pereira
Browse files

model_dir is important for evaluation

parent 6524d975
Branches
Tags
1 merge request!18Monitored training
...@@ -20,6 +20,7 @@ The configuration files should have the following objects totally: ...@@ -20,6 +20,7 @@ The configuration files should have the following objects totally:
## Required objects: ## Required objects:
model_dir
model_fn model_fn
eval_input_fn eval_input_fn
...@@ -27,7 +28,6 @@ The configuration files should have the following objects totally: ...@@ -27,7 +28,6 @@ The configuration files should have the following objects totally:
eval_interval_secs eval_interval_secs
run_once run_once
model_dir
run_config run_config
model_params model_params
steps steps
...@@ -114,12 +114,12 @@ def main(argv=None): ...@@ -114,12 +114,12 @@ def main(argv=None):
config_files = args['<config_files>'] config_files = args['<config_files>']
config = read_config_file(config_files) config = read_config_file(config_files)
model_dir = config.model_dir
model_fn = config.model_fn model_fn = config.model_fn
eval_input_fn = config.eval_input_fn eval_input_fn = config.eval_input_fn
eval_interval_secs = getattr(config, 'eval_interval_secs', 300) eval_interval_secs = getattr(config, 'eval_interval_secs', 300)
run_once = getattr(config, 'run_once', False) run_once = getattr(config, 'run_once', False)
model_dir = getattr(config, 'model_dir', None)
run_config = getattr(config, 'run_config', None) run_config = getattr(config, 'run_config', None)
model_params = getattr(config, 'model_params', None) model_params = getattr(config, 'model_params', None)
steps = getattr(config, 'steps', None) steps = getattr(config, 'steps', None)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment