search

iakovosevdaimon / Neural-Graph-Generator

Neural Graph Generator: Feature-Conditioned Graph Generation using Latent Diffusion Models
12 stars 4 forks source link

Any plan to release the well-trained model? #3

Closed Tinyyuumi closed 1 month ago

Tinyyuumi commented 1 month ago

Very good project. I see the AE model also need to train, so any plan to release well-trian model about AE? I can just retrain diffusion model. And which component is used for the best model, AE or VAE? Thanks!

iakovosevdaimon commented 1 month ago

Thanks! We used VAE instead of AE as it showed better performance during our experiment. In any case, we don't intend to release a pretrained AE.

Have a nice day!