Fix ParametricBuffer to consider __getitem__ fields (such as task lab…

…els)
ContinualAI · Dec 21, 2023 · 5749629 · 5749629
1 parent 0aec47e
commit 5749629
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/avalanche/training/storage_policy.py b/avalanche/training/storage_policy.py
@@ -461,7 +461,10 @@ def update(self, strategy: "SupervisedTemplate", **kwargs):
         self.update_from_dataset(strategy, new_data)
 
     def update_from_dataset(self, strategy, new_data):
-        self.buffer = self.buffer.concat(new_data)
+        if len(self.buffer) == 0:
+            self.buffer = new_data
+        else:
+            self.buffer = self.buffer.concat(new_data)
         self.resize(strategy, self.max_size)
 
     def resize(self, strategy, new_size: int):