diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml index cf3d680..2b69702 100644 --- 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