fix hasattr check for task-awareness

ContinualAI · Jun 3, 2024 · 4dedf6e · 4dedf6e
1 parent 91841ac
commit 4dedf6e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/avalanche/training/templates/base_sgd.py b/avalanche/training/templates/base_sgd.py
@@ -456,9 +456,7 @@ def make_train_dataloader(
             other_dataloader_args["ffcv_args"] = kwargs["ffcv_args"]
 
         # use task-balanced dataloader for task-aware benchmarks
-        if hasattr(self.experience, "task_label") or hasattr(
-            self.experience, "task_labels"
-        ):
+        if hasattr(self.experience, "task_labels"):
             self.dataloader = TaskBalancedDataLoader(
                 self.adapted_dataset,
                 oversample_small_groups=True,

diff --git a/avalanche/training/templates/problem_type/supervised_problem.py b/avalanche/training/templates/problem_type/supervised_problem.py
@@ -45,9 +45,7 @@ def criterion(self):
     def forward(self):
         """Compute the model's output given the current mini-batch."""
         # use task-aware forward only for task-aware benchmarks
-        if hasattr(self.experience, "task_labels") or hasattr(
-            self.experience, "task_label"
-        ):
+        if hasattr(self.experience, "task_labels"):
             return avalanche_forward(self.model, self.mb_x, self.mb_task_id)
         else:
             return self.model(self.mb_x)