From 548242e7a4fc7411b970c0a6ea65f69011b9cab0 Mon Sep 17 00:00:00 2001
From: Patrick Virie <p.virie@gmail.com>
Date: Thu, 16 Jan 2025 20:09:20 +0700
Subject: [PATCH] continue course

---
 tasks/rl_hopper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tasks/rl_hopper.py b/tasks/rl_hopper.py
index 390c014..5c6828d 100644
--- 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 = 4
 
     if course >= num_courses:
         logging.info("Experiment already completed")
@@ -283,7 +283,7 @@ def prepare_data_tuples(states, actions, rewards, num_layers, skip_steps):
         random.seed(random_seed)
         
         total_steps = 0
-        num_trials = 4000
+        num_trials = 2000
         print_steps = max(1, num_trials // 100)
         epsilon = 0.8 - 0.5 * (course + 1) / num_courses