diff_traj: bump batch size

d4l3k · Jul 24, 2024 · 49334d8 · 49334d8
1 parent cd11be9
commit 49334d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/diff_traj.py b/configs/diff_traj.py
@@ -20,6 +20,6 @@
     autolabel_path=None,  # "/mnt/ext3/autolabel2",
     mask_path="n/a",  # only used for rice dataset
     num_workers=16,
-    batch_size=4,
+    batch_size=8,
     autolabel=False,
 )