hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
hofee 2025-01-05 23:57:43 +08:00
commit fca984e76b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
configs/server/server_train_config.yaml
View File

@ -7,7 +7,7 @@ runner:
parallel: False parallel: False
experiment: experiment:
name: train_ab_global_only_with_accept_probability name: train_ab_global_only_with_wp_p++_dense
root_dir: "experiments" root_dir: "experiments"
use_checkpoint: False use_checkpoint: False
epoch: -1 # -1 stands for last epoch epoch: -1 # -1 stands for last epoch
@ -80,7 +80,7 @@ dataset:
pipeline: pipeline:
nbv_reconstruction_pipeline: nbv_reconstruction_pipeline:
modules: modules:
pts_encoder: pointnet_encoder pts_encoder: pointnet++_encoder
seq_encoder: transformer_seq_encoder seq_encoder: transformer_seq_encoder
pose_encoder: pose_encoder pose_encoder: pose_encoder
view_finder: gf_view_finder view_finder: gf_view_finder
@ -98,6 +98,7 @@ module:
pointnet++_encoder: pointnet++_encoder:
in_dim: 3 in_dim: 3
params_name: dense
transformer_seq_encoder: transformer_seq_encoder:
embed_dim: 256 embed_dim: 256