Is the pretrained encoder converted from tensorflow model or trained by pytorch?

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

Is the pretrained encoder converted from tensorflow model or trained by pytorch? #13

Open xukai8 opened 3 years ago

ShenYujun commented 3 years ago

It is currently converted from Tensorflow model, but we will release a PyTorch version training script in the genforce repo in the future. Please stay tuned.

xukai8 commented 3 years ago

@ShenYujun Thank you for your reply. Could you please release the code of converting the Tensorflow model to Pytorch model?

ShenYujun commented 3 years ago

You can find it from HiGAN and make simple modifications to support different w codes for different layers.