diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml
index 0fe2fcc..b346c21 100644
--- a/configs/server/server_train_config.yaml
+++ b/configs/server/server_train_config.yaml
@@ -39,7 +39,7 @@ dataset:
     type: train
     cache: True
     ratio: 1
-    batch_size: 64
+    batch_size: 32
     num_workers: 16
     pts_num: 8192
     load_from_preprocess: True
@@ -55,7 +55,7 @@ dataset:
     eval_list:
       - pose_diff
     ratio: 1
-    batch_size: 64
+    batch_size: 32
     num_workers: 12
     pts_num: 8192
     load_from_preprocess: True
@@ -71,7 +71,7 @@ dataset:
     eval_list:
       - pose_diff
     ratio: 1
-    batch_size: 64
+    batch_size: 32
     num_workers: 12
     pts_num: 8192
     load_from_preprocess: True
@@ -92,12 +92,12 @@ module:
 
   pointnet_encoder:
     in_dim: 3
-    out_dim: 1024
+    out_dim: 512
     global_feat: True
     feature_transform: False
 
   transformer_seq_encoder:
-    embed_dim: 1280
+    embed_dim: 768
     num_heads: 4
     ffn_dim: 256
     num_layers: 3