wip

2025-09-09 14:15:16 +02:00
parent ed80faf1e2
commit 86d9d96ca4
12 changed files with 725 additions and 14 deletions
--- a/Deep-SAD-PyTorch/src/main.py
+++ b/Deep-SAD-PyTorch/src/main.py
@@ -898,7 +898,7 @@ def main(
                    device=device,
                    n_jobs_dataloader=cfg.settings["n_jobs_dataloader"],
                    k_fold_idx=fold_idx,
-                    batch_size=256,
+                    batch_size=32,
                )

            retest_output_path = load_model / "retest_output"