icar test

ContinualAI · Oct 17, 2023 · f786641 · f786641
1 parent 8626323
commit f786641
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/training/test_strategies.py b/tests/training/test_strategies.py
@@ -851,7 +851,8 @@ def test_expertgate(self):
 
     def test_icarl(self):
         model, optimizer, criterion, benchmark = self.init_scenario(multi_task=False)
-        optimizer = SGD(model.parameters(), lr=1e-5)
+        model = SimpleMLP(input_size=6, hidden_size=10)
+        optimizer = SGD(model.parameters(), lr=.000001)
 
         strategy = ICaRL(
             model.features,
@@ -860,11 +861,11 @@ def test_icarl(self):
             20,
             buffer_transform=None,
             fixed_memory=True,
-            train_mb_size=10,
+            train_mb_size=32,
             train_epochs=2,
             eval_mb_size=50,
             device=self.device,
-            eval_every=1,
+            eval_every=-1,
         )
 
         run_strategy(benchmark, strategy)