diff --git a/examples/04_training/07_train_normalizing_flow.py b/examples/04_training/07_train_normalizing_flow.py index 1de4b349b..baa3eec85 100644 --- 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", )