ChenXi/FCP_gym
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename

Browse Source
This commit is contained in:
ChenXi
2025-11-21 02:28:20 -05:00
parent eaaa5519bd
commit a5bbf94917
5 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/gyms/Kick_20M.py
View File

@@ -25,7 +25,7 @@ Learn how to run forward using step primitive
'''
class Mimic(gym.Env):
class Kick_20M(gym.Env):
def __init__(self, ip, server_p, monitor_p, r_type, enable_draw) -> None:
self.phase = 0
self.init_step = 40
@@ -275,7 +275,7 @@ class Train(Train_Base):
# --------------------------------------- Run algorithm
def init_env(i_env):
def thunk():
return Mimic(self.ip, self.server_p + i_env, self.monitor_p_1000 + i_env, self.robot_type, False)
return Kick_20M(self.ip, self.server_p + i_env, self.monitor_p_1000 + i_env, self.robot_type, False)
return thunk
@@ -308,7 +308,7 @@ class Train(Train_Base):
def test(self, args):
# Uses different server and monitor ports
server = Server(self.server_p - 1, self.monitor_p, 1)
env = Mimic(self.ip, self.server_p - 1, self.monitor_p, self.robot_type, True)
env = Kick_20M(self.ip, self.server_p - 1, self.monitor_p, self.robot_type, True)
model = PPO.load(args["model_file"], env=env)
try: