From 830d51fc80c5422755ddc93126c8209031bf39c0 Mon Sep 17 00:00:00 2001 From: hofee Date: Tue, 29 Oct 2024 17:01:37 +0000 Subject: [PATCH] upd --- configs/server/server_train_config.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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