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_5M.py
View File

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