search

tensorflow / model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
https://www.tensorflow.org/model_optimization
Apache License 2.0
1.48k stars 320 forks source link

Is it possible to use this tool to optimize (quantize) a model trained with PyTorch? #1134

Open zhuoran-guo opened 2 weeks ago

zhuoran-guo commented 2 weeks ago

hello, I want to ask Is it possible to use this tool to optimize (quantize) a model trained with PyTorch or it only allowed the model trained with Keras ?

Thank you so much