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

add better defaults to TaskParameters

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-12 19:51:40 +00:00
committed by zach dwiel
parent 5758c2f23e
commit 61ed6b8ce4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/base_parameters.py
View File

@@ -415,8 +415,8 @@ 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="./experiments/test/", seed=None, save_checkpoint_secs=None): experiment_path='/tmp', seed=None, save_checkpoint_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
@@ -426,7 +426,7 @@ class TaskParameters(Parameters):
: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
self.task_index = None # TODO: not really needed self.task_index = 0 # TODO: not really needed
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