search

caio-freitas / GraphDiffusionImitate

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

Fix known dimensions during diffusion #114

Closed caio-freitas closed 6 months ago

caio-freitas commented 7 months ago

Add loss_mask = ~condition_mask