Here is a list of all functions with links to the classes they belong to:

• __init__() : train.VQVAETrainer, vqvae.VectorQuantizer, vqvae.VQVAE
• forward() : vqvae.VectorQuantizer, vqvae.VQVAE
• get_dataloader() : train.VQVAETrainer
• train() : train.VQVAETrainer