caio-freitas / GraphDiffusionImitate

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

Fix positional encodings #62

Closed caio-freitas closed 10 months ago

caio-freitas commented 10 months ago

In the DiffusionOrderingNetwork, forward function. The positional encoding should be added to the input encodings of the GAT instead of the output embeddings