Transformer based vision colab example

real-stanford / diffusion_policy

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

https://diffusion-policy.cs.columbia.edu/

MIT License

1.09k stars 203 forks source link

Transformer based vision colab example #54

Open StephenYangjz opened 3 months ago

StephenYangjz commented 3 months ago

Hi, I am wondering how much harder is the transformer-based diffusion policy is to train? Will it be possible to adapt it to the vision based colab example on colab? @NeilNie @pointW @jingxixu @cheng-chi

pengzhenghao commented 3 months ago

StephenYangjz commented 3 months ago

Just wanna kindly bump up the message as others are on the same boat. Any points are appreciated! @NeilNie @pointW @jingxixu @cheng-chi

simensaxe commented 3 months ago