vqvae.VectorQuantizer Class Reference

Inheritance diagram for vqvae.VectorQuantizer:

Public Member Functions
	__init__ (self, num_embeddings=512, embedding_dim=64, commitment_cost=0.25)
	forward (self, x)

Public Attributes
	embedding = nn.Embedding(num_embeddings, embedding_dim)
	commitment_cost = commitment_cost

Detailed Description

Definition at line 5 of file vqvae.py.

Constructor & Destructor Documentation

__init__()

vqvae.VectorQuantizer.__init__	(		self,
			num_embeddings = 512,
			embedding_dim = 64,
			commitment_cost = 0.25 )

Definition at line 6 of file vqvae.py.

Member Function Documentation

forward()

vqvae.VectorQuantizer.forward	(		self,
			x )

Definition at line 12 of file vqvae.py.

Member Data Documentation

commitment_cost

vqvae.VectorQuantizer.commitment_cost = commitment_cost

Definition at line 10 of file vqvae.py.

embedding

vqvae.VectorQuantizer.embedding = nn.Embedding(num_embeddings, embedding_dim)

Definition at line 8 of file vqvae.py.

---

The documentation for this class was generated from the following file:

• physaug/vqvae/vqvae.py