MODEL#

Classes#

class models.coda_prompt_utils.model.CodaPrompt(emb_d, n_tasks, prompt_param, key_dim=768)[source]#

Bases: Module

forward(x_querry, l, x_block, train=False, task_id=None)[source]#

process_task_count()[source]#

class models.coda_prompt_utils.model.Model(num_classes=10, pt=False, prompt_param=None)[source]#

Bases: Module

forward(x, pen=False, train=False)[source]#

Functions#

models.coda_prompt_utils.model.ortho_penalty(t)[source]#

models.coda_prompt_utils.model.tensor_prompt(a, b, c=None, ortho=False)[source]#