batch_size=8

juglab · Apr 18, 2024 · 10e6e96 · 10e6e96
1 parent f0ee330
commit 10e6e96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/denoisplit/notebooks/example.ipynb b/denoisplit/notebooks/example.ipynb
@@ -196,7 +196,7 @@
     "# change the training parameters\n",
     "config.training.lr = 3e-3\n",
     "config.training.max_epochs = 10\n",
-    "config.training.batch_size = 16\n",
+    "config.training.batch_size = 8\n",
     "config.training.num_workers = 4\n",
     "\n",
     "config.workdir = \".\""