upd

configs/server/server_train_config.yaml

@@ -7,9 +7,9 @@ runner:
     parallel: False
     
   experiment:
-    name: train_ab_global_only
+    name: train_ab_global_only_with_accept_probability
     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