VIOLA

VIOLA copied to clipboard

Question about the action Token and image augmentation

1

action_token_out = transformer_out[:, :, 0, :]. Hello, i don't know why directly take the first dimension of the output as the action_token_out. After your grouping, the grouped input should follow...

Tschian

real robot data collection

1

Do you have the scripts to collect the demos on the real robot?

HM102

Hello! Thanks for releasing this great works! I am trying to reproduce this model in the real-world, so i use the viola dataset. I was wondering what is the scale...

Nimolty

Cannot import name 'CropRandomizer' from 'robomimic.models.base_nets'

3

I try training the model with your generated datasets, and get this error: **Cannot import name 'CropRandomizer' from 'robomimic.models.base_nets'** With suspecting incorrect installation of _robomimic_, I go to https://github.com/ARISE-Initiative/robomimic/blob/master/robomimic/models/base_nets.py to...

BaymaxShao

How to visualize when running the program viola_bc/final_eval_script.py? Where does the offscreen_visualization function in img = offscreen_visualization(env, use_eye_in_hand=cfg.algo.use_eye_in_hand) come from? I didn't find this function

jhcharlotte

Thanks for releasing your great works!! Can you share Real-Robot Dataset's experiment config file(maybe xx_viola.yaml..) and checkpoint .pth file?

dohu337

Is the architecture training different policies for different tasks? If not how are we specifying the task at test time?

Williambert