PNN#

Arguments#

Classes#

class models.pnn.Pnn(backbone, loss, args, transform, dataset=None)[source]#

Bases: ContinualModel

Progressive Neural Networks.

COMPATIBILITY: List[str] = ['task-il']#
NAME: str = 'pnn'#
end_task(dataset)[source]#
forward(x, task_label)[source]#
observe(inputs, labels, not_aug_inputs, epoch=None)[source]#

Functions#

models.pnn.get_pnn_backbone(bone, old_cols=None, x_shape=None)[source]#