pytorch-vqvae
pytorch-vqvae copied to clipboard
loss
i want know the code of loss: log_px = nll.mean().item() - np.log(128) + kl_d.item() in that code of loss , the 128 of 'np.log(128)' is value of Z_DIM ?