Wipe task

Open Nimingez opened this issue 2 years ago • 0 comments

I collect my wipe datase from robosuite using spacemouse to train with robomimic

python dataset_states_to_obs.py --dataset /home/xxx/robomimic/demo.hdf5 --output_name image_dense.hdf5 --done_mode 2 --camera_names agentview robot0_eye_in_hand --camera_height 84 --camera_width 84 I want to train with bc_transformer

python train.py --config ../config/default_templates/bc_transformer.json --dataset /home/xxx/robomimic/image.hdf5

RuntimeError: cannot reshape tensor of 0 elements into shape [-1, 0] because the unspecified dimension size -1 can be any value and is ambiguous

Mar 10 '24 08:03 Nimingez