We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5141c19 commit 7ed8ac9Copy full SHA for 7ed8ac9
README.md
@@ -17,10 +17,10 @@ import torch
17
from vector_quantize_pytorch import VectorQuantize
18
19
vq = VectorQuantize(
20
- dim = 256,
21
- n_embed = 512, # size of the dictionary
22
- decay = 0.8, # the exponential moving average decay, lower means the dictionary will change faster
23
- commitment = 1. # the weight on the commitment loss
+ dim = 256,
+ n_embed = 512, # size of the dictionary
+ decay = 0.8, # the exponential moving average decay, lower means the dictionary will change faster
+ commitment = 1. # the weight on the commitment loss
24
)
25
26
x = torch.randn(1, 1024, 256)
0 commit comments