Input labels in train_vae.py

wiseodd / controlled-text-generation

Reproducing Hu, et. al., ICML 2017's "Toward Controlled Generation of Text"

BSD 3-Clause "New" or "Revised" License

242 stars 63 forks source link

Closed nabihach closed 6 years ago

nabihach commented 6 years ago

Line 59 in train_vae.py: inputs, labels = dataset.next_batch(args.gpu)

Here, the inputs are the sentences and labels are sentiments. However, this is just a VAE, so shouldn't the labels be the same as the inputs?

wiseodd commented 6 years ago

You are correct. That's why, if you notice, labels is not being used in train_vae.py at all. It is being used in train_discriminator.py.

nabihach commented 6 years ago

Ah right. Thanks!