bug fix: commitment loss factor beta

Author: wcshin-git

taming/modules/vqvae/quantize.py

@@ -66,7 +66,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