I want to get the“hidden_states” data in the Bert model #20664
Labels
feature request
request for unsupported feature or enhancement
model:transformer
issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc.
Describe the feature request
I'm running a TTS project where the Bert model needs to get the“hidden_states” component of the middle layer output for later processing. I'm using C + + and can't seem to find a way to get the middle layer output, any suggestions?
Describe scenario use case
This method is handy for using the Bert Model, because hidden_states are also useful features, which can be implemented in the Python version of Transformer, but is difficult to implement in C + + .
The text was updated successfully, but these errors were encountered: