Enum RerankHead 

#[non_exhaustive]
pub enum RerankHead {
    BertPooler,
    RobertaClassificationHead,
    LogitScore {
        true_token_id: u32,
        false_token_id: u32,
        unembedding: Unembedding,
    },
}

Which rerank head a checkpoint runs, serialized with a kind tag such as {"kind": "bert_pooler"}.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

BertPooler

tanh(Linear(hidden, hidden)) over the first position, then Linear(hidden, 1), from the checkpoint’s bert.pooler.dense and classifier.

RobertaClassificationHead

The same arithmetic from classifier.dense and classifier.out_proj, with no pooler tensor.

LogitScore

The logit of true_token_id minus that of false_token_id at the prompt’s last position: the Qwen3 rerankers.

Fields

true_token_id: u32

false_token_id: u32

unembedding: Unembedding

Trait Implementations

impl Clone for RerankHead

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for RerankHead

impl Debug for RerankHead

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for RerankHead

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for RerankHead

impl PartialEq for RerankHead

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl Serialize for RerankHead

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for RerankHead