search

Nrgeup / controllable-text-attribute-transfer

Code for "Controllable Unsupervised Text Attribute Transfer via Editing Entangled Latent Representation" (NeurIPS 2019)
Apache License 2.0
128 stars 27 forks source link

Usage of torch.nn.GRU #17

Open MokkeMeguru opened 4 years ago

MokkeMeguru commented 4 years ago

torch.nn.GRU 's input shape is [sequence_size, batch_size, hidden_size]. But your code such as https://github.com/Nrgeup/controllable-text-attribute-transfer/blob/master/method/mymodel-yelp/model2.py#L315 is [batch_size, sequence_size, hidden_size]. I think your code is something wrong...

Nrgeup commented 3 years ago

Thank you for your reminder!