dualprompt#

DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning

Note

WARNING: DualPrompt USES A CUSTOM BACKBONE: vit_base_patch16_224. The backbone is a ViT-B/16 pretrained on Imagenet 21k and finetuned on ImageNet 1k.

Classes

class models.dualprompt.DualPrompt(backbone, loss, args, transform)[source]#

Bases: ContinualModel

COMPATIBILITY: List[str] = ['class-il', 'task-il']#
NAME: str = 'dualprompt'#
begin_task(dataset)[source]#
forward(x)[source]#
get_parameters()[source]#
static get_parser()[source]#
Return type:

ArgumentParser

observe(inputs, labels, not_aug_inputs, epoch=None)[source]#