Large-Trajectory-Model / ATM

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

problem about training the track transformer #4

Open KaiTang98 opened 1 week ago

KaiTang98 commented 1 week ago

Hello, thank you for presenting this interesting work. I am attempting to reproduce the training based on the Libero dataset.

However, I've noticed that when training the track transformer, the train_dataset_list and val1_dataset_list obtained through:

train_dataset_list = glob(os.path.join(root_dir, f"{suite_name}//train/")) val1_dataset_list = glob(os.path.join(root_dir, f"{suite_name}//val/"))

are empty. I'm wondering if the data preprocessing stage requires additional processing of the data obtained from preprocess_libero.py? Many thanks!

image

image

KaiTang98 commented 1 week ago

I checked the dataset inside linero_goal as an example. The dataset structure is shown as follows. -- libero_goal ------ open_the_middle_drawer.... ----------images ----------videos ----------demo_0.hdf5 ----------demo_1.hdf5 ----------....hdf5 ----------env_meta.json

KaiTang98 commented 1 week ago

I realized that I just need to create these two folders by myself... Now I can train the model...😂😂

caichuang0415 commented 6 days ago

I realized that I just need to create these two folders by myself... Now I can train the model...😂😂

Thank you very much I have just come with this problem as well 😂😂