Does anyone try to apply this project to BERT?
https://github.com/google-research/bert Thank you very much!
Not that we know of. I don't believe the open source bert trainer uses Keras. It would beb nice to see somebody trying it out though.
@haozha111 is trying this out
@haozha111
how to apply this api to bert? do you train bert by keras? can we use bert trained by tensorflow?
The pruning API only supports keras model right now. I'm just starting to trying it out, and don't have any code examples now. Will post here once I get some good results. Thanks!
Hi @haozha111, is there some updates?
The pruning API only supports keras model right now. I'm just starting to trying it out, and don't have any code examples now. Will post here once I get some good results. Thanks!
Hi @haozha111, is there some updates?
no, I haven't tried it recently. if you are interested, please follow docs in https://www.tensorflow.org/model_optimization to apply it to BERT, also cc'ing Jaesung who is currently working on MOT.