diff --git a/examples/ur_sim/env.py b/examples/ur_sim/env.py index ffd36c3..9ed8c23 100644 --- a/examples/ur_sim/env.py +++ b/examples/ur_sim/env.py @@ -33,6 +33,7 @@ simulation_app = app_launcher.app import cv2 import h5py import torch +# torch.set_printoptions(precision=3, threshold=10, edgeitems=3) import gymnasium import numpy as np @@ -115,7 +116,13 @@ class URSimEnvironment(_environment.Environment): action = action.copy() action[-1] = action[-1] * 2 - 1 + ##### + # action = np.zeros(7) + # action[-1] = -1 + #### + action = torch.tensor(action, dtype=torch.float32)[None] + # print(action) gym_obs, reward, terminated, truncated, info = self._gym.step(action) diff --git a/examples/ur_sim/main.py b/examples/ur_sim/main.py index 97a4d9b..3915965 100644 --- a/examples/ur_sim/main.py +++ b/examples/ur_sim/main.py @@ -19,7 +19,7 @@ class Args: task: str = "PIBussing" seed: int = 0 - action_horizon: int = 10 + action_horizon: int = 16 host: str = "0.0.0.0" port: int = 8000