ColinKohler/BulletArm

single process bugs

Opened this issue · 0 comments

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