gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +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):
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
@@ -430,7 +430,7 @@ class TaskParameters(Parameters):
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