Large-Trajectory-Model / ATM

Official codebase for "Any-point Trajectory Modeling for Policy Learning"
https://xingyu-lin.github.io/atm/
MIT License
61 stars 7 forks source link

Regarding direct training deployment on real robot #1

Open yolo01826 opened 1 month ago

yolo01826 commented 1 month ago

Dear Author,

I really like your work and have a few questions I'd like to ask you.

  1. Firstly, if the keypoint tracking method is trained on a single task with 100 human demonstrations and 10 robot teleoperations, can it yield a well-performing tracking model?
  2. Does the keypoint information passed into the policy consist of its pixel coordinates, or does it include some form of rotation?
  3. The paper mentions using 4 A100 GPUs for training. Is it possible to train a good single-task policy using a 4090 GPU instead?

Thank you.

AlvinWen428 commented 1 month ago

Hi, thanks for your interest in our work.

  1. Yes. A single task makes it easier to learn a good track transformer.
  2. Only 2D coordinates are fed into the policy. Because there are multiple point tracks, the policy will infer the rotation by itself.
  3. Yes, but you need to tune the hyperparameters.

I hope my answers will be helpful.

dblanm commented 1 month ago

Hey! I want to follow on questions about training and deployment in real.

  1. Is there any example of the data required for providing real-world demonstrations with a robot?
  2. Will you provide code for the real-world experiments with the UR5?
  3. How are you sending the commands with the UR5, is this ROS, RTDE with movep and movej, or something different?
  4. Related to 3, Could you explain the pipeline: ATM tracker & policy + send commands to robots
  5. Could you also specify more in detail what is needed for doing this with real hardware: what inputs for doing this with real hardware, requirements in camera viewpoints, any required transformations of observations to the policy, calibrations, limits of robot/human demonstration velocities, GPUs required for doing inference, etc.

Thanks for the help!