update config

3dlg-hcvc · Apr 6, 2024 · 2319057 · 2319057
1 parent 3487f0a
commit 2319057
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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