Closed zjrn closed 4 years ago
I find all bert model is a single layer in final Keras Model. How can I get each layer's output in Bert Model?
you can use the out_layer_ndxs parameter, which is a list specifying the layer outputs BertModelLayer will return.
out_layer_ndxs
BertModelLayer
I find all bert model is a single layer in final Keras Model. How can I get each layer's output in Bert Model?