guangyliu / LatentOps

Source code of LatentOps
77 stars 9 forks source link

Add more instructions to train vae/classifer/gan? #3

Closed dhx20150812 closed 1 year ago

dhx20150812 commented 1 year ago

Hi, @guangyliu , that's really an interesting and helpful work.

Recently, I'm trying to conduct experiments on my own dataset(agnews), but I find it difficult for me to run your code. So it would be great if you can add more docs about how to train the vae/classifer/gan.

Here are something I think might be important and useful:

  1. Dataset preparation, including data format and detailed hyperparameters.
  2. Traing scripts. For example, I tried to train my own latent classifer using the script code/train_classifier_latent.sh, but I'm not sure whether it is correct. Furthermore, I cannot find any scripts to train the gan. Maybe I missed something important.

Thanks in advance! Looking forward to your reply.

guangyliu commented 1 year ago

Thanks for the comments. I will give more details about the training part in the next two days.

guangyliu commented 1 year ago

Please see the updated version in https://github.com/guangyliu/LatentOps#train-vae