pass dataloader kwargs to run_validation

during refactor, did not pass these (incl batch_size and num_workers) to the method run_validation, resulting in use of batch_size=1 and num_workers=0 during validations in SpectrogramClassifier.train() loop
kitzeslab · Sep 13, 2024 · 7a2ea1b · 7a2ea1b
1 parent 3118648
commit 7a2ea1b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/opensoundscape/ml/cnn.py b/opensoundscape/ml/cnn.py
@@ -1565,7 +1565,12 @@ def train(
             if epoch % validation_interval == 0:
                 if validation_df is not None:
                     self._log("\nValidation.")
-                    val_metrics = self.run_validation(validation_df)
+                    val_metrics = self.run_validation(
+                        validation_df,
+                        batch_size=batch_size,
+                        num_workers=num_workers,
+                        raise_errors=raise_errors,
+                    )
                     self.valid_metrics[self.current_epoch] = val_metrics
                     score = val_metrics[self.score_metric]  # overall score
                     if wandb_session is not None: