finish partial_global inference

This commit is contained in:
2024-11-26 15:40:00 +08:00
parent 2c8ef20321
commit be835aded4
3 changed files with 67 additions and 14 deletions

View File

@@ -6,7 +6,7 @@ runner:
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: train_ab_global_only
name: train_ab_partial
root_dir: "experiments"
epoch: -1 # -1 stands for last epoch
@@ -15,7 +15,7 @@ runner:
- OmniObject3d_test
blender_script_path: "/media/hofee/data/project/python/nbv_reconstruction/blender/data_renderer.py"
output_dir: "/media/hofee/data/data/new_inference_test_output"
output_dir: "/media/hofee/data/data/new_partial_inference_test_output"
pipeline: nbv_reconstruction_pipeline
voxel_size: 0.003
min_new_area: 1.0
@@ -66,7 +66,7 @@ module:
global_feat: True
feature_transform: False
transformer_seq_encoder:
embed_dim: 256
embed_dim: 320
num_heads: 4
ffn_dim: 256
num_layers: 3