ContinualAlgorithm¶

class cl_gym.algorithms.ContinualAlgorithm(backbone, benchmark, params, requires_memory=False)[source]¶

Bases: object

Base class for continual learning algorithms.
It contains abstractions for implementing different algorithms, and also implementations shared among all algorithms.
It can be used for Naive(Finetune) algorithm or Stable SGD algorithm by Mirzadeh et. al.

prepare_criterion(task_id)[source]¶

prepare_optimizer(task_id)[source]¶

prepare_train_loader(task_id)[source]¶

prepare_validation_loader(task_id)[source]¶

sample_batch_from_memory()[source]¶

setup()[source]¶

teardown()[source]¶

training_epoch_end()[source]¶

training_step(task_ids, inp, targ, optimizer, criterion)[source]¶

training_step_end()[source]¶

training_task_end()[source]¶

update_episodic_memory()[source]¶

Algorithms AGEM