PhysAugNet 1.0.1
VQ-VQE powered augmentation for metal defect segmentation
Loading...
Searching...
No Matches
vqvae.VectorQuantizer Member List

This is the complete list of members for vqvae.VectorQuantizer, including all inherited members.

__init__(self, num_embeddings=512, embedding_dim=64, commitment_cost=0.25)vqvae.VectorQuantizer
commitment_costvqvae.VectorQuantizer
embeddingvqvae.VectorQuantizer
forward(self, x)vqvae.VectorQuantizer