Update configs

Catchxu · Jun 3, 2024 · f3f3389 · f3f3389
1 parent 3372538
commit f3f3389
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/stands/configs.py b/src/stands/configs.py
@@ -101,8 +101,8 @@ def __init__(self, z_dim):
         self.KMeans_n_init = 20
         self.cross_attn = True
         self.TFBlock = {
-            'g_dim': z_dim,
-            'p_dim': z_dim,
+            'g_dim': z_dim*2,
+            'p_dim': z_dim*2,
             'num_layers': 3,
             'nheads': 8,
             'hidden_dim': 512, 
@@ -111,6 +111,6 @@ def __init__(self, z_dim):
 
         # training process
         self.learning_rate = 1e-4
-        self.n_epochs = 20000,
+        self.n_epochs = 20000
         self.update_interval = 10
         self.weight_decay = 1e-4