Will you release a pytorch version of the code to train the encoder?

genforce / idinvert_pytorch

[ECCV 2020] In-Domain GAN Inversion for Real Image Editing (PyTorch code)

https://genforce.github.io/idinvert/

MIT License

407 stars 54 forks source link

Will you release a pytorch version of the code to train the encoder? #11

Closed doutzi closed 3 years ago

doutzi commented 3 years ago

Hi, I was wondering if you plan to release a PyTorch version of the code necessary to train the encoder? That would be great.

ShenYujun commented 3 years ago

Yes. Please keep an eye on the genforce repo.

Hsintien-Ng commented 3 years ago

I have reproduced the train_encoder.py myself according to the tf version. I found an interesting thing that the learning rate in pytorch version should be smaller than 1e-4 while the one in tf version could be 1e-4. Also, the results of training with lr=1e-5 seems not as good as that in tf version.

doutzi commented 3 years ago

Hi @Hsintien-Ng, could you possibly share your reproduced train_encoder.py? That would be very nice :) I'm not familiar with tf and therefore I don't feel confident to reproduce it myself, but I'd very much like to experiment with the encoder training.

Hsintien-Ng commented 3 years ago

Yep, @doutzi, please follow the reproduced project in https://github.com/Hsintien-Ng/idinvert_pytorch-reproduced. If there exists some errors, please point out them.