update config

3dlg-hcvc · Apr 2, 2024 · dc3f916 · dc3f916
1 parent 2dd510c
commit dc3f916
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/cage.yaml b/configs/cage.yaml
@@ -13,7 +13,7 @@ system:
     K: ${system.K}
     batch_size: 64
     hash_table: indexes/hash_table.json # for pred mode
-    pred_mode: cond_graph 
+    pred_mode: cond_graph # cond_graph or cond_type
     pred_n_samples: 10 # the number of samples to generate for each condition
 
   model:
@@ -46,7 +46,7 @@ checkpoint:
 
 trainer:
   fast_dev_run: false
-  max_epochs: 5000
+  max_epochs: 30000
   log_every_n_steps: 500
   num_sanity_val_steps: 0
   limit_train_batches: 1.0