search

ColinKohler / BulletArm

BulletArm is a benchmark and learning-environment for robotic manipulation.
MIT License
69 stars 18 forks source link

single process bugs #13

Open tinzzyli opened 1 year ago

tinzzyli commented 1 year ago

when run /bulletarm_baselines/fc_dqn/scripts/main.py with num_process = 0, some errors relate to tensor shape occurs. For example:

Traceback (most recent call last): File "bulletarm_baselines/fc_dqn/scripts/main.py", line 288, in train() File "bulletarm_baselines/fc_dqn/scripts/main.py", line 157, in train planner_actions_star_idx, planner_actions_star = agent.getActionFromPlan(plan_actions) File "/BulletArm/bulletarm_baselines/fc_dqn/agents/agents_3d/base_3d.py", line 20, in getActionFromPlan x = plan[:, 0:1] IndexError: too many indices for tensor of dimension 1

here plan_actions.shape is torch.Size([4]) while expected to be torch.Size([1, 4]) in getActionFromPlan