CL-Gym 1.0.0-beta documentation
Quickstart
How does CL-Gym work?
CL-Gym Components
Benchmarks
Benchmark
ContinualMNIST
RotatedMNIST
PermutedMNIST
SplitCIFAR10
SplitCIFAR100
Toy2DClassification
Toy1DRegression
PAMAP2
Backbones
ContinualBackbone
MLP2Layers
CNN1D
ResNet18Small
Trainer
ContinualTrainer
Algorithms
ContinualAlgorithm
AGEM
ERRingBuffer
EWC
OGD
MCSGD
Multitask
Callbacks
ContinualCallback
ModelCheckpoint
MetricCollector
ExperimentManager
Index
A
|
B
|
C
|
E
|
F
|
G
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
A
AGEM (class in cl_gym.algorithms)
B
Benchmark (class in cl_gym.benchmarks)
blocks (cl_gym.backbones.MLP2Layers attribute)
(cl_gym.backbones.ResNet18Small attribute)
C
calculate_line_loss() (cl_gym.algorithms.MCSGD method)
calculate_point_loss() (cl_gym.algorithms.MCSGD method)
CNN1D (class in cl_gym.backbones)
connect() (cl_gym.utils.callbacks.ContinualCallback method)
ContinualAlgorithm (class in cl_gym.algorithms)
ContinualBackbone (class in cl_gym.backbones)
ContinualCallback (class in cl_gym.utils.callbacks)
ContinualMNIST (class in cl_gym.benchmarks.mnist)
ContinualTrainer (class in cl_gym.trainer)
E
ERRingBuffer (class in cl_gym.algorithms)
EWC (class in cl_gym.algorithms)
ExperimentManager (class in cl_gym.utils.callbacks)
F
find_connected_minima() (cl_gym.algorithms.MCSGD method)
fit() (cl_gym.trainer.ContinualTrainer method)
forward() (cl_gym.backbones.CNN1D method)
(cl_gym.backbones.ContinualBackbone method)
(cl_gym.backbones.MLP2Layers method)
G
get_block_grads() (cl_gym.backbones.ContinualBackbone method)
(cl_gym.backbones.MLP2Layers method)
get_block_outputs() (cl_gym.backbones.ContinualBackbone method)
(cl_gym.backbones.MLP2Layers method)
get_block_params() (cl_gym.backbones.ContinualBackbone method)
(cl_gym.backbones.MLP2Layers method)
L
load() (cl_gym.benchmarks.Benchmark method)
load_datasets() (cl_gym.benchmarks.Benchmark method)
(cl_gym.benchmarks.mnist.ContinualMNIST method)
(cl_gym.benchmarks.PAMAP2 method)
(cl_gym.benchmarks.Toy1DRegression method)
(cl_gym.benchmarks.Toy2DClassification method)
load_joint() (cl_gym.benchmarks.Benchmark method)
load_memory() (cl_gym.benchmarks.Benchmark method)
load_memory_joint() (cl_gym.benchmarks.Benchmark method)
load_subset() (cl_gym.benchmarks.Benchmark method)
log_metrics() (cl_gym.utils.callbacks.MetricCollector method)
M
MCSGD (class in cl_gym.algorithms)
MetricCollector (class in cl_gym.utils.callbacks)
MLP2Layers (class in cl_gym.backbones)
ModelCheckpoint (class in cl_gym.utils.callbacks)
multi_head (cl_gym.backbones.MLP2Layers attribute)
(cl_gym.backbones.ResNet18Small attribute)
Multitask (class in cl_gym.algorithms)
N
num_classes_per_head (cl_gym.backbones.MLP2Layers attribute)
(cl_gym.backbones.ResNet18Small attribute)
O
OGD (class in cl_gym.algorithms)
on_after_fit() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.MetricCollector method)
on_after_setup() (cl_gym.utils.callbacks.ContinualCallback method)
on_after_teardown() (cl_gym.utils.callbacks.ContinualCallback method)
on_after_training_epoch() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.MetricCollector method)
(cl_gym.utils.callbacks.ModelCheckpoint method)
on_after_training_step() (cl_gym.utils.callbacks.ContinualCallback method)
on_after_training_task() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.MetricCollector method)
(cl_gym.utils.callbacks.ModelCheckpoint method)
on_before_fit() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.ExperimentManager method)
(cl_gym.utils.callbacks.MetricCollector method)
(cl_gym.utils.callbacks.ModelCheckpoint method)
on_before_setup() (cl_gym.utils.callbacks.ContinualCallback method)
on_before_teardown() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.ExperimentManager method)
on_before_training_epoch() (cl_gym.utils.callbacks.ContinualCallback method)
on_before_training_step() (cl_gym.utils.callbacks.ContinualCallback method)
on_before_training_task() (cl_gym.utils.callbacks.ContinualCallback method)
(cl_gym.utils.callbacks.MetricCollector method)
P
PAMAP2 (class in cl_gym.benchmarks)
PermutedMNIST (class in cl_gym.benchmarks)
plot_metrics() (cl_gym.utils.callbacks.MetricCollector method)
precompute_joint_indices() (cl_gym.benchmarks.Benchmark method)
precompute_memory_indices() (cl_gym.benchmarks.Benchmark method)
(cl_gym.benchmarks.mnist.ContinualMNIST method)
(cl_gym.benchmarks.PAMAP2 method)
(cl_gym.benchmarks.Toy1DRegression method)
(cl_gym.benchmarks.Toy2DClassification method)
precompute_seq_indices() (cl_gym.benchmarks.Benchmark method)
precompute_subset_indices() (cl_gym.benchmarks.Benchmark method)
prepare_criterion() (cl_gym.algorithms.ContinualAlgorithm method)
prepare_datasets() (cl_gym.benchmarks.Benchmark method)
prepare_optimizer() (cl_gym.algorithms.ContinualAlgorithm method)
prepare_train_loader() (cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.algorithms.Multitask method)
prepare_validation_loader() (cl_gym.algorithms.ContinualAlgorithm method)
R
ResNet18Small (class in cl_gym.backbones)
RotatedMNIST (class in cl_gym.benchmarks)
run() (cl_gym.trainer.ContinualTrainer method)
S
sample_batch_from_memory() (cl_gym.algorithms.ContinualAlgorithm method)
sample_uniform_class_indices() (cl_gym.benchmarks.Benchmark method)
sanity_check_inputs() (cl_gym.benchmarks.Benchmark method)
sanity_check_transforms() (cl_gym.benchmarks.Benchmark method)
save_metrics() (cl_gym.utils.callbacks.MetricCollector method)
select_output_head() (cl_gym.backbones.ContinualBackbone method)
setup() (cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.trainer.ContinualTrainer method)
SplitCIFAR10 (class in cl_gym.benchmarks)
SplitCIFAR100 (class in cl_gym.benchmarks)
T
teardown() (cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.trainer.ContinualTrainer method)
Toy1DRegression (class in cl_gym.benchmarks)
Toy2DClassification (class in cl_gym.benchmarks)
train_algorithm_on_task() (cl_gym.trainer.ContinualTrainer method)
training (cl_gym.backbones.CNN1D attribute)
(cl_gym.backbones.ContinualBackbone attribute)
(cl_gym.backbones.MLP2Layers attribute)
(cl_gym.backbones.ResNet18Small attribute)
training_epoch_end() (cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.algorithms.MCSGD method)
training_step() (cl_gym.algorithms.AGEM method)
(cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.algorithms.ERRingBuffer method)
(cl_gym.algorithms.EWC method)
(cl_gym.algorithms.MCSGD method)
(cl_gym.algorithms.OGD method)
training_step_end() (cl_gym.algorithms.ContinualAlgorithm method)
training_task_end() (cl_gym.algorithms.ContinualAlgorithm method)
(cl_gym.algorithms.EWC method)
(cl_gym.algorithms.MCSGD method)
(cl_gym.algorithms.OGD method)
U
update_episodic_memory() (cl_gym.algorithms.ContinualAlgorithm method)
V
validate_algorithm_on_task() (cl_gym.trainer.ContinualTrainer method)