wangyuxinwhy / uniem

unified embedding model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

What hidden states should I use?

zhangzhenyu13 opened this issue · comments

commented

the output of encoder is N*d states for each text input;

  1. simcse use the 1st one as the output.
  2. the sentence-transformers use the mean pooling(only for attention mask=1 states).

Your default is the same as mean pooling?

yes, use the mean pooling(only for attention mask=1 states)