From 23190578134ef3b70faf24c37da62ca5c1deb942 Mon Sep 17 00:00:00 2001
From: SevenLJY <jiayeeliu@gmail.com>
Date: Sat, 6 Apr 2024 16:03:38 -0700
Subject: [PATCH] update config

---
 configs/cage.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/cage.yaml b/configs/cage.yaml
index ab580fd..925cac0 100644
--- a/configs/cage.yaml
+++ b/configs/cage.yaml
@@ -11,7 +11,7 @@ system:
     split_file: indexes/data_split.json
     augment: true
     K: ${system.K}
-    batch_size: 128
+    batch_size: 64 # each gpu has 64 samples
     hash_table: indexes/hash_table.json # for pred mode
     pred_mode: cond_graph # cond_graph, cond_type, cond_axis, cond_box, cond_label
     pred_n_samples: 10 # the number of samples to generate for each condition