Merge branch 'ab_global_only' of http://git.hofee.top/hofee/nbv_reconstruction into ab_global_only

This commit is contained in:
hofee 2025-01-05 23:57:43 +08:00
commit fca984e76b

View File

@ -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