yxuansu / TaCL

[NAACL'22] TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning

Home Page:https://arxiv.org/abs/2111.04198

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

`AttributeError: 'str' object has no attribute 'squeeze'`

mathshangw opened this issue · comments

excuse me if i need to get embeddings = encoded_layers[11].squeeze(0)
but with using your model i got

AttributeError: 'str' object has no attribute 'squeeze'