Merge pull request #1637 from Chillthrower/master

PNN adapter arguement missing in the _init_() method #1625
ContinualAI · May 9, 2024 · 0d6f715 · 0d6f715
2 parents 5bc33e1 + 9637aa0
commit 0d6f715
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/avalanche/models/pnn.py b/avalanche/models/pnn.py
@@ -20,6 +20,7 @@ def __init__(self, in_features, out_features_per_column, num_prev_modules):
         :param num_prev_modules: number of previous modules
         """
         super().__init__()
+        self.num_prev_modules = num_prev_modules
         # Eq. 1 - lateral connections
         # one layer for each previous column. Empty for the first task.
         self.lat_layers = nn.ModuleList([])