Open WenBingo opened 1 year ago
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 ?
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 ?