increase batch size

RasmusOrsoe · Aug 6, 2024 · dd41659 · dd41659
1 parent 7f472cf
commit dd41659
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/04_training/07_train_normalizing_flow.py b/examples/04_training/07_train_normalizing_flow.py
@@ -209,7 +209,7 @@ def main(
         "gpus",
         ("max-epochs", 1),
         "early-stopping-patience",
-        ("batch-size", 16),
+        ("batch-size", 50),
         "num-workers",
     )