use cuda as device in training

caio-freitas / GraphDiffusionImitate

Diffusion-based graph generative policies for imitation learning in robotics tasks 🧠🤖

MIT License

1 stars 1 forks source link

Closed caio-freitas closed 11 months ago