-
Notifications
You must be signed in to change notification settings - Fork 8
Open
Description
Hi,
I want to ask how I can get the corresponding "ee_states" in the dataset from the online rollout.
In the dataset, "data/demo_{episode_idx}/obs/ee_states" has the shape of 6;
In online rollout, obs = env.set_init_state(init_states_), obs["robot0_eef_pos"] has the shape of 3, and obs["robot0_eef_quat"] has the shape of 4.
Does "ee_states" includes both "robot0_eef_pos" and "robot0_eef_quat"?
If yes, why there's one more dimension? Which dimension should I abandon?
If no, how can I get "ee_states" from obs?
Thanks!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels