This repository was archived by the owner on Feb 7, 2025. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 111
VQ-VAE LDM tutorial
#496
Hi everyone,
I am trying to test the 2D LDM tutorial with the AEKL, as in https://github.com/Project-MONAI/GenerativeModels/blob/main/tutorials/generative/2d_ldm/2d_ldm_tutorial.ipynb . I was wondering if there is an available version of this tutorial (basically just the U-net training part of the LDM) using MONAI's VQ-VAE, as I have not been able to find one. Thanks in advance.
Guido
All reactions
Replies: 1 comment
There isn't but the VQ-VAE and AE can be used very similarly so you should be able to just swap out the AE for a VQ-VAE, maybe using this for guidance on how to use the VQ-VAE if needed.
I also have a research codebase that makes use of MONAI gen to train an LDM with a VQ-VAE which may be useful here
All reactions
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment