deploy pointnet++ again

This commit is contained in:
2024-12-28 19:38:27 +00:00
parent 91cabec977
commit 47ea0ac434
29 changed files with 10 additions and 1720 deletions

View File

@@ -7,9 +7,9 @@ runner:
parallel: False
experiment:
name: train_ab_global_only
name: train_ab_global_only_pointnet++
root_dir: "experiments"
use_checkpoint: True
use_checkpoint: False
epoch: -1 # -1 stands for last epoch
max_epochs: 5000
save_checkpoint_interval: 1
@@ -96,6 +96,9 @@ module:
global_feat: True
feature_transform: False
pointnet++_encoder:
in_dim: 3
transformer_seq_encoder:
embed_dim: 256
num_heads: 4