search

cindyxinyiwang / deep-latent-sequence-model

Pytorch implementation of "A Probabilistic Formulation of Unsupervised Text Style Transfer" by He. et. al. at ICLR 2020
163 stars 26 forks source link

Cant find intial reconstruction loss code #4

Closed drumilT closed 4 years ago

drumilT commented 4 years ago

The paper mentions that you used initially for k epochs, a reconstruction loss. However I could find it, could you point me to it please?

jxhe commented 4 years ago

https://github.com/cindyxinyiwang/deep-latent-sequence-model/blob/8bbc71ff2bf856cafcf97a3174cf4869a21a9eff/src/utils.py#L75

Here the noise_loss is the reconstruction loss mentioned in the paper