issues
search
caio-freitas
/
GraphDiffusionImitate
Diffusion-based graph generative policies for imitation learning in robotics tasks 🧠🤖
MIT License
1
stars
1
forks
source link
Implement ARGD
#46
Closed
caio-freitas
closed
11 months ago
caio-freitas
commented
1 year ago
[x] Implement most simplified model, be able to train and evaluate it
[x] Get architecture of denoising network as similar as possible to the one described in paper
[x] Add/Create Diffusion Ordering Network (DON), as in paper
[ ] Get architecture of DON as similar as possible to paper
[x] Get loss functions as similar as possible as the ones in paper
[x] Train DON with REINFORCE alongside with the denoising network, as in paper
[x] Fully connect graph with masked edges
[x] Keep only node to be denoised, among the masked next, during the generation step