caio-freitas / GraphDiffusionImitate

Diffusion-based graph generative policies for imitation learning in robotics tasks 🧠🤖
MIT License
1 stars 1 forks source link

Implement Base Train Script #12

Closed caio-freitas closed 1 year ago

caio-freitas commented 1 year ago

Depending on the used models, the train script can be significantly changed. E.g. diffusion models.