From 450d9f965ac0e9d0b27bac9d621e393de03cd510 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Thu, 6 Oct 2022 09:08:56 +0200
Subject: [PATCH] update default batch size for cgan training
---
mu_map/training/cgan.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mu_map/training/cgan.py b/mu_map/training/cgan.py
index aa93a31..c9a6b9b 100644
--- a/mu_map/training/cgan.py
+++ b/mu_map/training/cgan.py
@@ -304,7 +304,7 @@ if __name__ == "__main__":
parser.add_argument(
"--batch_size",
type=int,
- default=8,
+ default=64,
help="the batch size used for training",
)
parser.add_argument(
--
GitLab