gan compression on tensorflow

[sepideh-srj]

Hi,

I wanted to implement GAN compression on TensorFlow so I can use it on a mobile application.
I'm pretty new to this field and I'm not aware of the challenges of doing something like this might have.
Should I implement the whole thing on Tensorflow myself? would it be as fast as it is in Pytorch?
Or should I convert the pytorch model to tensorflow with onnx?
I would really appreciate any help

[lmxyy]

Cool! I think you could first convert the pytorch model to tensorflow with onnx first. I am not familiar with tensorflow, so currently I could not predict what kind of challenges you may meet. If you have any questions, feel free to open issues here.

[funnyQZQ]

@sepideh-srj As far as i know, you can train and compress your model with Pytorch and than convert the Pytorch model to tensorflow or tflite model, it will be the simplest way.