0

I want to remove the last layer(s) from MobileBERT from Hub. I know there is a solution for Keras Model in TensorFlow, but this case is different from that one.

I was thinking of something like this, but it doesn't seem user-friendly.

What is the common way of doing this?

oguz ismail
  • 1
  • 16
  • 47
  • 69
Andrew Ishutin
  • 99
  • 6

1 Answers1

1

There is no first-class APIs to do this. The solution along the lines what you have mentioned is the way to go.

Andrey Khorlin
  • 213
  • 1
  • 4