Merge branch 'master' of https://github.com/wcshin-git/taming-transformers-forked into wcshin-git-master

Author: pesser

taming/modules/vqvae/quantize.py

@@ -70,7 +70,7 @@ def forward(self, z):
         # ......\end......... (TODO)
 
         # compute loss for embedding
-        loss = torch.mean((z_q.detach()-z)**2) + self.beta * \
+        loss = self.beta * torch.mean((z_q.detach()-z)**2) + \
             torch.mean((z_q - z.detach()) ** 2)
 
         # preserve gradients