Merge pull request #1668 from AndreaCossu/master

Preparation for 0.6
ContinualAI · Oct 29, 2024 · b34abe5 · b34abe5
2 parents 56f2b9f + 09477f3
commit b34abe5
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
+
+