Changed train batch size in example to 1024

PreferredAI · Oct 11, 2023 · 60c1641 · 60c1641
1 parent 1c44609
commit 60c1641
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ngcf_example.py b/examples/ngcf_example.py
@@ -41,7 +41,7 @@
     layer_sizes=[64, 64, 64],
     dropout_rates=[0.1, 0.1, 0.1],
     early_stopping={"min_delta": 1e-4, "patience": 50},
-    train_batch_size=256,
+    train_batch_size=1024,
     learning_rate=0.001,
     lambda_reg=1e-5,
     verbose=True,