increase episode length

PVirie · Jan 15, 2025 · f3b8f4f · f3b8f4f
1 parent 200076c
commit f3b8f4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tasks/rl_hopper.py b/tasks/rl_hopper.py
@@ -162,7 +162,7 @@ def setup():
 def train(context, parameter_path):
 
     course = context.course
-    num_courses = 2
+    num_courses = 10
 
     if course >= num_courses:
         logging.info("Experiment already completed")
@@ -304,7 +304,7 @@ def prepare_data_tuples(states, actions, rewards, num_layers, skip_steps):
             states = []
             actions = []
             rewards = []
-            for _ in range(250):
+            for _ in range(400):
                 if random.random() <= epsilon or course == 0:
                     selected_action = env.action_space.sample()
                     # quantize