real-stanford / diffusion_policy

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
https://diffusion-policy.cs.columbia.edu/
MIT License
1.1k stars 206 forks source link

How to generate low dim embeddings the for Kitchen Envs #25

Closed xavihart closed 8 months ago

xavihart commented 9 months ago

In the dataset, the observation sequence is sized N*60, is there any code snippets to generate the low dim embeddings?

cheng-chi commented 8 months ago

Hi @xavihart I'm not quite sure I understand your question. The native observation space for the kitchen task is 60 (though a lot of these dimensions are empty), and the dataset is directly extracted from *.mjl files published in the original relay-policy-learining repo inside KitchenMjlLowdimDataset

xavihart commented 8 months ago

I got it, so it is not vision-based?

cheng-chi commented 8 months ago

Yes, we only have state-based version of the kitchen task in this repo.