From 9a33c795a20e090a750b1b9739454edcc0dda578 Mon Sep 17 00:00:00 2001 From: retinfai Date: Thu, 7 Mar 2024 15:14:33 +1300 Subject: [PATCH] Add number steps per train policy config --- cares_reinforcement_learning/util/configurations.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cares_reinforcement_learning/util/configurations.py b/cares_reinforcement_learning/util/configurations.py index 190696e8..0521ad85 100644 --- a/cares_reinforcement_learning/util/configurations.py +++ b/cares_reinforcement_learning/util/configurations.py @@ -29,6 +29,7 @@ class TrainingConfig(SubscriptableClass): number_steps_per_evaluation: Optional[int] = 10000 number_eval_episodes: Optional[int] = 10 + number_steps_per_train_policy: Optional[int] = 1 plot_frequency: Optional[int] = 100 checkpoint_frequency: Optional[int] = 100