Can I use MorphNet to slim my model trained by Pytorch?

google-research / morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure

Apache License 2.0

1.02k stars 148 forks source link

Can I use MorphNet to slim my model trained by Pytorch? #174

Closed LeiWang1999 closed 4 years ago

LeiWang1999 commented 4 years ago

Just like the title，I have a model trained by Pytorch, Can i use this project to slim it.. Eg: Convert Pytorch's model to ONNX, then read by Tensorflow? Thanks!

eladeban commented 4 years ago

sorry fo the late reply, you need to do the structure learning in TensorFlow as this is a TensorFlow framework... it is not about converting the trained model, its about learning the structure. Implementing the algorithm int Pytorch should not be super hard, and you are welcome to add it here..