Episode ends values to learn tactile representations

ZhengtongXu / UniT

UniT: Unified Tactile Representation for Robot Learning

MIT License

27 stars 3 forks source link

Episode ends values to learn tactile representations #2

Open Jcastanyo opened 4 hours ago

Jcastanyo commented 4 hours ago

Hi,

I am trying to train your model to learn tactile representations with data from my sensor. I see that in you replay_buffer.zarr data, you are also saving episode ends. Is it necessary if I'm only interesed in the tactile representation task? If I just save one number that matches the shape of my dataset, can I train the model without problems?

Thanks in advance.

ZhengtongXu commented 2 hours ago

Yes. One episode is basically a chunk of data with shape (T, data_shape). Here T could be the total number of your images and data_shape is the size of your images.