rail-berkeley / crossformer

MIT License
138 stars 11 forks source link

Demo for the navigation task #3

Open TimeEscaper opened 1 week ago

TimeEscaper commented 1 week ago

Dear authors,

Thank you very much for releasing the code of the great work! We are now trying to deploy your model for the mobile robot navigation task. We are using inference_pretrained.ipynb as a reference and trying to adapt it for the navigation demo, but facing some troubles with that. Can you please outline the steps that should be done for running your model in navigation mode?

Thank you in advance!

HomerW commented 6 days ago

To adapt the inference example from the notebook for navigation:

However, note that the full trajectory inference example in the notebook doesn't actually handle the timestep pad mask correctly. To actually roll out the policy on a robot you can use the evaluation server here. This is how we evaluated on mobile navigation robots. This server has a function stack_and_pad that will correctly stack the history of observations and create a correct timestep padding mask. Hope this helps!