Skip to content
This repository has been archived by the owner on Dec 11, 2022. It is now read-only.

Commit

Permalink
fix renaming: save_checkpoint_sec -> checkpoint_save_secs
Browse files Browse the repository at this point in the history
  • Loading branch information
zach-nervana committed Oct 24, 2018
1 parent 78cf25c commit f835ac9
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions rl_coach/base_parameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -416,21 +416,21 @@ def path(self):

class TaskParameters(Parameters):
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 evaluate_only: the task will be used only for evaluating the model
: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 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
"""
self.framework_type = framework_type
self.task_index = 0 # TODO: not really needed
self.evaluate_only = evaluate_only
self.use_cpu = use_cpu
self.experiment_path = experiment_path
self.save_checkpoint_secs = save_checkpoint_secs
self.checkpoint_save_secs = checkpoint_save_secs
self.seed = seed


Expand Down
2 changes: 1 addition & 1 deletion rl_coach/coach.py
Original file line number Diff line number Diff line change
Expand Up @@ -350,7 +350,7 @@ def main():
experiment_path=args.experiment_path,
seed=args.seed,
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__)

start_graph(graph_manager=graph_manager, task_parameters=task_parameters)
Expand Down

0 comments on commit f835ac9

Please sign in to comment.