upd cluster inference

This commit is contained in:
2025-01-05 15:50:04 +00:00
parent 88d44f020e
commit 1535a48a3f
3 changed files with 166 additions and 2 deletions

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