From 2487039445473c236afbcf758d19ab3f4557b2d3 Mon Sep 17 00:00:00 2001 From: hofee Date: Tue, 29 Oct 2024 12:18:51 +0000 Subject: [PATCH] global_only: config --- configs/server/server_train_config.yaml | 21 +-------------------- core/pipeline.py | 1 - 2 files changed, 1 insertion(+), 21 deletions(-) diff --git a/configs/server/server_train_config.yaml b/configs/server/server_train_config.yaml index 190e6e3..721f327 100644 --- a/configs/server/server_train_config.yaml +++ b/configs/server/server_train_config.yaml @@ -78,7 +78,7 @@ dataset: pipeline: - nbv_reconstruction_local_pts_pipeline: + nbv_reconstruction_pipeline: modules: pts_encoder: pointnet_encoder seq_encoder: transformer_seq_encoder @@ -87,25 +87,6 @@ pipeline: eps: 1e-5 global_scanned_feat: True - nbv_reconstruction_global_pts_pipeline: - modules: - pts_encoder: pointnet_encoder - pose_seq_encoder: transformer_seq_encoder - pose_encoder: pose_encoder - view_finder: gf_view_finder - eps: 1e-5 - global_scanned_feat: True - - nbv_reconstruction_global_pts_n_num_pipeline: - modules: - pts_encoder: pointnet_encoder - transformer_seq_encoder: transformer_seq_encoder - pose_encoder: pose_encoder - view_finder: gf_view_finder - pts_num_encoder: pts_num_encoder - eps: 1e-5 - global_scanned_feat: True - module: diff --git a/core/pipeline.py b/core/pipeline.py index 797fd87..ce2d755 100644 --- a/core/pipeline.py +++ b/core/pipeline.py @@ -29,7 +29,6 @@ class NBVReconstructionPipeline(nn.Module): self.eps = float(self.config["eps"]) - self.enable_global_scanned_feat = self.config["global_scanned_feat"] def forward(self, data): mode = data["mode"]