PROMPT

Classes

class models.l2p_utils.prompt.Prompt(length=5, embed_dim=768, embedding_key='mean', prompt_init='uniform', prompt_pool=False, prompt_key=False, pool_size=None, top_k=None, batchwise_prompt=False, prompt_key_init='uniform', prompt_shuffle=False)

Bases: Module

forward(x_embed, prompt_mask=None, cls_features=None)

l2_normalize(x, dim=None, epsilon=1e-12)

Normalizes a given vector or matrix.