pytorch/examples

VAE reconstruction loss (BCE)

開放

#460 建立於 2018年12月1日

 (2 則留言) (3 個反應) (0 位負責人)Python (9,429 個分叉)batch import
good first issue

倉庫指標

星標
 (21,634 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

The current implementation uses

F.binary_cross_entropy(recon_x, x.view(-1, 784), reduction='sum')

as the reconstruction loss. The image x has pixel values in [0,1]. This is not the same as Bernoulli log likelihood. The images would have to binarized.

In Ladder Variational Autoencoders by Sonderby et al, they binarize the images as a Bernoulli sample after each epoch.

貢獻者指南