search

Large-Trajectory-Model / ATM

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

Some Issues on ATM Model Training. #19

Open cuite1999 opened 1 month ago

cuite1999 commented 1 month ago

ATM is really a very delicate work, which is very rewarding. I encountered some problems in the process of reproducing the results of the ATM paper. I fully trained the track model and the action model according to the method and process you provided. However, whether I used the entire dataset training data, or used 20% of the training data to train the action model, or used the tracking transformer model you provided, plus 20% of the data training action model, the results during testing could not reach the results in the paper. Even the success rate of many tasks is 0. There must be something wrong with my reproduction process. Can you share your training or testing log, or the final model you obtained by training under different data set partitions? Sorry to bother you.

AlvinWen428 commented 1 month ago

Hello, thanks for your nice words. The policy checkpoints trained with 20% data have been provided in README. I wonder whether you have tried to evaluate them and how is the performance. I'm so sorry that I didn't save the training logs.