gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

fix renaming: save_checkpoint_sec -> checkpoint_save_secs

Browse Source
This commit is contained in:
zach dwiel
2018-10-24 10:52:18 -04:00
parent 78cf25c09a
commit f835ac902c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/base_parameters.py
View File

@@ -416,13 +416,13 @@ class AgentParameters(Parameters):
class TaskParameters(Parameters): class TaskParameters(Parameters):
def __init__(self, framework_type: str='tensorflow', evaluate_only: bool=False, use_cpu: bool=False, def __init__(self, framework_type: str='tensorflow', evaluate_only: bool=False, use_cpu: bool=False,
experiment_path='/tmp', seed=None, save_checkpoint_secs=None): experiment_path='/tmp', seed=None, checkpoint_save_secs=None):
""" """
:param framework_type: deep learning framework type. currently only tensorflow is supported :param framework_type: deep learning framework type. currently only tensorflow is supported
:param evaluate_only: the task will be used only for evaluating the model :param evaluate_only: the task will be used only for evaluating the model
:param use_cpu: use the cpu for this task :param use_cpu: use the cpu for this task
:param experiment_path: the path to the directory which will store all the experiment outputs :param experiment_path: the path to the directory which will store all the experiment outputs
:param save_checkpoint_secs: the number of seconds between each checkpoint saving :param checkpoint_save_secs: the number of seconds between each checkpoint saving
:param seed: a seed to use for the random numbers generator :param seed: a seed to use for the random numbers generator
""" """
self.framework_type = framework_type self.framework_type = framework_type
@@ -430,7 +430,7 @@ class TaskParameters(Parameters):
self.evaluate_only = evaluate_only self.evaluate_only = evaluate_only
self.use_cpu = use_cpu self.use_cpu = use_cpu
self.experiment_path = experiment_path self.experiment_path = experiment_path
self.save_checkpoint_secs = save_checkpoint_secs self.checkpoint_save_secs = checkpoint_save_secs
self.seed = seed self.seed = seed

2
rl_coach/coach.py
View File

@@ -350,7 +350,7 @@ def main():
experiment_path=args.experiment_path, experiment_path=args.experiment_path,
seed=args.seed, seed=args.seed,
use_cpu=args.use_cpu, use_cpu=args.use_cpu,
save_checkpoint_secs=args.save_checkpoint_secs) checkpoint_save_secs=args.checkpoint_save_secs)
task_parameters.__dict__ = add_items_to_dict(task_parameters.__dict__, args.__dict__) task_parameters.__dict__ = add_items_to_dict(task_parameters.__dict__, args.__dict__)
start_graph(graph_manager=graph_manager, task_parameters=task_parameters) start_graph(graph_manager=graph_manager, task_parameters=task_parameters)