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?
I was training with real UR5 through main.py.
but, it occurred the error like following :
it didn't match the dimenstion. but, I don't guess what is problem exactly. Who you know me which one should change?
Thanks.