From 61ed6b8ce4208393ac050d6818469c94ebf9b1d5 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Wed, 12 Sep 2018 19:51:40 +0000 Subject: [PATCH] add better defaults to TaskParameters --- rl_coach/base_parameters.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rl_coach/base_parameters.py b/rl_coach/base_parameters.py index de737eb..1bbb53a 100644 --- a/rl_coach/base_parameters.py +++ b/rl_coach/base_parameters.py @@ -415,8 +415,8 @@ class AgentParameters(Parameters): class TaskParameters(Parameters): - 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): + def __init__(self, framework_type: str='tensorflow', evaluate_only: bool=False, use_cpu: bool=False, + experiment_path='/tmp', seed=None, save_checkpoint_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 @@ -426,7 +426,7 @@ class TaskParameters(Parameters): :param seed: a seed to use for the random numbers generator """ 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.use_cpu = use_cpu self.experiment_path = experiment_path