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

Function of src/transformer.py #6

Closed drumilT closed 4 years ago

drumilT commented 4 years ago

I see that the code has an option of using transformer based model however the transformer model architecture hasn't been imported from src/transformer.py. If incomplete could you please direct to parts of code that need "attention".

jxhe commented 4 years ago

The codebase is originated from a machine translation codebase, and the transformer part in this repo was never tested, thus we don't recommend digging into the transformer part.