-
Notifications
You must be signed in to change notification settings - Fork 11
Open
Labels
enhancementNew feature or requestNew feature or request
Description
Я говорю про то чтобы не использовать _classifier_hook можно просто сделать SentenceTranformer, а не CrossEncoder
причины почему мы используем хук:
* это работает :) * нам нужно использовать один трансформер в двух режимах в зависимости от флага `train_classifier`: с претренированной головой и со своей кастомнойПретренированную голову точно можно ипользовать с помощью
CrossEncoder.predict(), а о том что ее можно использовать и с помощью какого-то метода классаSentenceTransformerмне неизвестно. Если это и можно сделать, то наверное это что-то экзотическое и не задокументировано в самой библиотекеsentence_transformers
Да, нельзя использовать доп голову. Просто сейчас у нас реализован получение эмбеда из 0 токена (предположим, что это CLS) но не все модели обучались с ним и не факт, что это хорошо работать будет
Originally posted by @Samoed in #88 (comment)
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request