vineetjohn / linguistic-style-transfer

Neural network parametrized objective to disentangle and transfer style and content in text
Apache License 2.0
138 stars 33 forks source link

Fix problem of poor reconstruction accuracy #11

Closed vineetjohn closed 6 years ago

vineetjohn commented 6 years ago

Clip gradients

vineetjohn commented 6 years ago

Dropout embedded encoder and decoder inputs

vineetjohn commented 6 years ago

Add beam search decoder for inference

vineetjohn commented 6 years ago

Throttle learning rate EDIT: Done

vineetjohn commented 6 years ago

Pass encoder cell states to the decoder initial state This might bypass the disentanglement EDIT: Done

vineetjohn commented 6 years ago

Use temperature for variability Not applicable right now

vineetjohn commented 6 years ago

Switch to SGD from Adam Adam converges quicker but is prone to over-fitting EDIT: Done

vineetjohn commented 6 years ago

Add cost for repeated tokens? Not applicable right now

vineetjohn commented 6 years ago

Append hidden state to every time step Not applicable right now

vineetjohn commented 6 years ago

Increase RNN neurons EDIT: Done

vineetjohn commented 6 years ago

Reduce batch size to avoid over-generalization EDIT: Done

vineetjohn commented 6 years ago

Fixed. Solution was to reduce batch size.