search

s-tian / vp2

VP2 Benchmark (A Control-Centric Benchmark for Video Prediction, ICLR 2023)
24 stars 1 forks source link

Are the 5K trajectories in the training dataset provided collected from the same task? #2

Open Muksaki opened 1 year ago

Muksaki commented 1 year ago

With reference to the 5k_slice_rendered_256.hdf5 dataset, I'm wondering whether those 5k trajectories are collected from one same task or sliced from the 50K total dataset

s-tian commented 8 months ago

The 5K trajectories in the 5k_slice_rendered_256.hdf5 dataset are for the robosuite tabletop environment only, but contain data that interacts with each of the four objects (large cube, small cube, cylinder, sphere), so they include examples of all tasks in the robosuite environment. They are sliced from the 50K total dataset.