search

andyzeng / visual-pushing-grasping

Train robotic agents to learn to plan pushing and grasping actions for manipulation with deep reinforcement learning.
http://vpg.cs.princeton.edu/
BSD 2-Clause "Simplified" License
898 stars 315 forks source link

Backpropagation error #45

Open Geonhee-LEE opened 4 years ago

Geonhee-LEE commented 4 years ago

I was training with real UR5 through main.py.

but, it occurred the error like following :

Traceback (most recent call last): File "main.py", line 446, in main(args) File "main.py", line 298, in main trainer.backprop(prev_color_heightmap, prev_valid_depth_heightmap, prev_primitive_action, prev_best_pix_ind, label_value) File "/home/geonhee-ml/rl_ws/src/visual-pushing-grasping/trainer.py", line 335, in backprop loss_value = loss.cpu().data.numpy()[0] IndexError: too many indices for array

it didn't match the dimenstion. but, I don't guess what is problem exactly. Who you know me which one should change?

Thanks.