Hello. This doubt may seem a little bit less researched, but here I go. I wanted to convert the stylegan generator to tensorflow js model. But this I understand that stylegan python (using TF2.x) has custom loops and objects in the generator object. I followed the steps mentioned below :
I right now have a keras model stylegan generator with weights loaded and the class has a call function that takes in noise and gives images. The definition is same as “StyleGAN_G” class in StyleGAN-Tensorflow2/stylegan.py at master · ialhashim/StyleGAN-Tensorflow2 (github.com). The weights I loaded was using the tutorial given in the same repository StyleGAN-Tensorflow2/stylegan.ipynb at master · ialhashim/StyleGAN-Tensorflow2 (github.com). So which format do I save the generator object (before converting it to tfjs model) ? Tensorflow seems to have a lot of save options
And after saving, what all flag values I have to set tfjsconvertor ?