Revert "improve training speed and add speed constrain"

This reverts commit 648cf32e9c.
This commit is contained in:
2026-03-13 08:43:28 -04:00
parent 648cf32e9c
commit 3a42120857
4 changed files with 40 additions and 78 deletions

View File

@@ -18,18 +18,9 @@ class Server():
# makes it easier to kill test servers without affecting train servers
cmd = "rcssservermj"
for i in range(n_servers):
port = first_server_p + i
mport = first_monitor_p + i
server_cmd = f"{cmd} --aport {port} --mport {mport} --no-render --no-realtime"
self.rcss_processes.append(
subprocess.Popen(
server_cmd.split(),
stdout=subprocess.DEVNULL,
stderr=subprocess.STDOUT,
start_new_session=True
)
subprocess.Popen((f"{cmd} --aport {first_server_p+i} --mport {first_monitor_p+i}").split(),
stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT, start_new_session=True)
)
def check_running_servers(self, psutil, first_server_p, first_monitor_p, n_servers):
@@ -69,4 +60,4 @@ class Server():
def kill(self):
for p in self.rcss_processes:
p.kill()
print(f"Killed {self.n_servers} rcssservermj processes starting at {self.first_server_p}")
print(f"Killed {self.n_servers} rcssservermj processes starting at {self.first_server_p}")

View File

@@ -46,7 +46,7 @@ class Train_Base():
@staticmethod
def prompt_user_for_model(self):
gyms_logs_path = "./scripts/gyms/logs/"
gyms_logs_path = "./mujococodebase/scripts/gyms/logs/"
folders = [f for f in listdir(gyms_logs_path) if isdir(join(gyms_logs_path, f))]
folders.sort(key=lambda f: os.path.getmtime(join(gyms_logs_path, f)), reverse=True) # sort by modification date