include batch size as arg meta_evaluation.py

metaquantus/meta_evaluation.py

@@ -130,6 +130,7 @@ def __call__(
         channel_first: Optional[bool] = True,
         softmax: Optional[bool] = False,
         device: Optional[str] = None,
+        batch_size: Optional[int] = 64,
         model_predict_kwargs: Optional[Dict[str, Any]] = {},
         score_direction: Optional[str] = None,
     ):
@@ -179,6 +180,7 @@ def __call__(
             channel_first=channel_first,
             softmax=softmax,
             device=device,
+            batch_size=batch_size,
         )
 
         # Run inference.
@@ -352,6 +354,7 @@ def run_perturbation_analysis(
                         },
                         model_predict_kwargs=model_predict_kwargs,
                         softmax=softmax,
+                        batch_size=batch_size,
                         device=device,
                     )
 
@@ -397,6 +400,7 @@ def run_perturbation_analysis(
                         model_predict_kwargs=model_predict_kwargs,
                         softmax=softmax,
                         device=device,
+                        batch_size=batch_size,
                     )
 
                 self.results_eval_scores_perturbed[test_name][i] = scores_perturbed