diff --git a/avalanche/training/supervised/strategy_wrappers.py b/avalanche/training/supervised/strategy_wrappers.py index b13091b64..62a211bdd 100644 --- a/avalanche/training/supervised/strategy_wrappers.py +++ b/avalanche/training/supervised/strategy_wrappers.py @@ -1696,5 +1696,6 @@ def __init__( "MAS", "BiC", "MIR", + "PackNet", "FromScratchTraining", ] diff --git a/docs/conf.py b/docs/conf.py index c9c09b056..7af606792 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -401,6 +401,26 @@ "MetaLearningStrategyProtocol", "MultiParamTransformCallable", "TqdmUpTo", + "TaskAwareSupervisedClassificationDataset", + "HybridFfcvLoader", + "FeatureExtractorModel", + "FixedSizeExperienceSplitter", + "FeatureDataset", + "CyclicSampler", + "LazyTrainValSplitter", + "AvalancheImageNet", + "SingleDatasetDataLoader", + "TaskAwareClassificationDataset", + "TaskAware", + "Adaptable", + "FFCVInfo", + "PackNetPlugin", + "ClassesTimeline", + "Buffer", + "BoundaryAware", + "PackNetModule", + "Agent", + "PositionalArgumentsDeprecatedWarning", ] undocumented_classes_to_ignore = set(undocumented_classes_to_ignore) diff --git a/docs/models.rst b/docs/models.rst index a80abe459..18dbe83fa 100644 --- a/docs/models.rst +++ b/docs/models.rst @@ -27,6 +27,7 @@ networks, ...). MultiTaskModule IncrementalClassifier MultiHeadClassifier + CosineIncrementalClassifier Progressive Neural Networks ^^^^^^^^^^^^^^^^^^^^^^^^^^^ @@ -68,6 +69,13 @@ Models SlimResNet18 MTSlimResNet18 ExpertGate + packnet.PackNetModel + packnet.packnet_simple_mlp + FeCAMClassifier + cosine_layer.CosineLinear + cosine_layer.SplitCosineLinear + packnet.WeightAndBiasPackNetModule + Model Wrappers and Utilities diff --git a/docs/training.rst b/docs/training.rst index aeafa4ef1..256f27d24 100644 --- a/docs/training.rst +++ b/docs/training.rst @@ -83,9 +83,12 @@ Ready-to-use continual learning strategies. ER_ACE LearningToPrompt SCR + PackNet FromScratchTraining ExpertGateStrategy DER + ER_AML + FeatureReplay supervised.lamaml.LaMAML supervised.lamaml_v2.LaMAML @@ -132,6 +135,8 @@ Loss Functions LearningWithoutForgetting ACECriterion SCRLoss + MaskedCrossEntropy + AMLCriterion Training Plugins @@ -184,3 +189,12 @@ Strategy implemented as plugins in `avalanche.training.plugins`. MIRPlugin RARPlugin FromScratchTrainingPlugin + FeatureDistillationPlugin + FeCAMOracle + MemoryFeCAMUpdate + CurrentDataFeCAMUpdate + MemoryNCMUpdate + NCMOracle + CurrentDataNCMUpdate + +