upd

2024-10-29 17:01:37 +00:00
parent e81d6c9bd1
commit 830d51fc80
1 changed files with 2 additions and 2 deletions
--- a/configs/server/server_train_config.yaml
+++ b/configs/server/server_train_config.yaml
@@ -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