bench tests: update use_bias parameter for RPS calculators

deel-ai · Feb 16, 2024 · 07fbb55 · 07fbb55
1 parent f1a360f
commit 07fbb55
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/benchmark/test_bench.py b/tests/benchmark/test_bench.py
@@ -107,7 +107,8 @@ def test_rps_lje():
                                                             test_batch_size=10,
                                                             epochs_to_save=None,
                                                             take_batch=take_batch,
-                                                            verbose_training=False)
+                                                            verbose_training=False,
+                                                            use_bias=False)
 
     influence_factory = RPSLJEFactory('exact')
     result = cifar10_evaluator.evaluate(influence_factory=influence_factory, nbr_of_evaluation=2, verbose=False)
@@ -126,7 +127,8 @@ def test_rps_l2():
                                                             test_batch_size=10,
                                                             epochs_to_save=None,
                                                             take_batch=take_batch,
-                                                            verbose_training=False)
+                                                            verbose_training=False,
+                                                            use_bias=False)
 
     influence_factory = RPSL2Factory(CategoricalCrossentropy(from_logits=True, reduction=Reduction.NONE),
                                      lambda_regularization=10.0)