How to extract the word embedding

google-research / albert

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations

Apache License 2.0

3.23k stars 570 forks source link

Open 256785 opened 4 years ago

256785 commented 4 years ago

When I use the bert,I can extract the feature of word embedding like this [https://github.com/google-research/bert/blob/master/extract_features.py] But when I use the albert , I extract the wrong dim of the embedding , could you help me to extract the word embeding?

s0miya commented 3 years ago

are you able to extract the feature of word embedding now?