Tuning params for Pos/Dir envs

This commit is contained in:
Dominik Moritz Roth 2024-06-28 12:11:17 +02:00
parent 3864909174
commit fe90208307

View File

@ -21,8 +21,8 @@ class DirPointEnv(AgentModel):
MANUAL_COLLISION: bool = True MANUAL_COLLISION: bool = True
RADIUS: float = 0.4 RADIUS: float = 0.4
OBJBALL_TYPE: str = "hinge" OBJBALL_TYPE: str = "hinge"
VELOCITY_LIMITS: float = 0.333 VELOCITY_LIMITS: float = 0.15
ACCELERATION_LIMITS: float = 0.1 ACCELERATION_LIMITS: float = 0.005
def __init__(self, file_path: Optional[str] = None, maze_env=None) -> None: def __init__(self, file_path: Optional[str] = None, maze_env=None) -> None:
self.maze_env = maze_env self.maze_env = maze_env
@ -41,6 +41,9 @@ class DirPointEnv(AgentModel):
# Scale the action to the maze size # Scale the action to the maze size
direction = (action)/2.5 * maze_size direction = (action)/2.5 * maze_size
print('ooo')
self.data.qvel[:2]*=0.95
# Get current position and velocity # Get current position and velocity
current_vel = self.data.qvel[:2] current_vel = self.data.qvel[:2]