diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml index a3d8bec..87ba6e3 100644 --- a/configs/server/server_train_config.yaml +++ b/configs/server/server_train_config.yaml @@ -7,7 +7,7 @@ runner: parallel: False experiment: - name: train_ab_global_only_with_accept_probability + name: train_ab_global_only_with_wp_p++_dense root_dir: "experiments" use_checkpoint: False epoch: -1 # -1 stands for last epoch @@ -80,7 +80,7 @@ dataset: pipeline: nbv_reconstruction_pipeline: modules: - pts_encoder: pointnet_encoder + pts_encoder: pointnet++_encoder seq_encoder: transformer_seq_encoder pose_encoder: pose_encoder view_finder: gf_view_finder @@ -98,6 +98,7 @@ module: pointnet++_encoder: in_dim: 3 + params_name: dense transformer_seq_encoder: embed_dim: 256