diff --git a/rl_coach/training_worker.py b/rl_coach/training_worker.py index 9eb0898..e107ad8 100644 --- a/rl_coach/training_worker.py +++ b/rl_coach/training_worker.py @@ -25,7 +25,7 @@ def training_worker(graph_manager, checkpoint_dir, policy_type): # initialize graph task_parameters = TaskParameters() task_parameters.__dict__['checkpoint_save_dir'] = checkpoint_dir - task_parameters.__dict__['save_checkpoint_secs'] = 20 + task_parameters.__dict__['checkpoint_save_secs'] = 20 graph_manager.create_graph(task_parameters) # save randomly initialized graph