Accessing BERT encoding layer output.

utterworks / fast-bert

Super easy library for BERT based NLP models

Apache License 2.0

1.86k stars 341 forks source link

Open nectario opened 4 years ago

nectario commented 4 years ago

How can I access the BERT encoding layer output (not the prediction output)?

aaronbriel commented 4 years ago

You can access them in the model within fast-bert.

nectario commented 4 years ago

Thank you! Which file? Sorry I am a bit new to Pytorch too. So any code helpful code snippet would be appreciated.