runner:
  general:
    seed: 0
    device: cuda
    cuda_visible_devices: "1"
    parallel: False
    
  experiment:
<<<<<<< HEAD
    name: test_new_pipeline_train_overfit
=======
    name: debug
>>>>>>> 63a246c0c87d42f04076a459adcfdc88c954b09c
    root_dir: "experiments"
    use_checkpoint: False
    epoch: -1 # -1 stands for last epoch
    max_epochs: 5000
    save_checkpoint_interval: 1
    test_first: False  

  train:
    optimizer:
      type: Adam
      lr: 0.0001
    losses: 
      - gf_loss
    dataset: OmniObject3d_train
  test:
    frequency: 3 # test frequency
    dataset_list:
      #- OmniObject3d_test
      - OmniObject3d_val

  pipeline: nbv_reconstruction_global_pts_n_num_pipeline
  
dataset:
  OmniObject3d_train:
    root_dir: "/home/data/hofee/project/nbv_rec/data/sample_for_training_new"
    model_dir: "../data/scaled_object_meshes"
    source: nbv_reconstruction_dataset
    split_file: "/home/data/hofee/project/nbv_rec/data/sample.txt"
    type: train
    cache: True
    ratio: 1
    batch_size: 16
    num_workers: 16
    pts_num: 4096
    load_from_preprocess: True

  OmniObject3d_test:
    root_dir: "/home/data/hofee/project/nbv_rec/data/sample_for_training_new"
    model_dir: "../data/scaled_object_meshes"
    source: nbv_reconstruction_dataset
    split_file: "/home/data/hofee/project/nbv_rec/data/sample.txt"
    type: test
    cache: True
    filter_degree: 75
    eval_list:
      - pose_diff
    ratio: 0.05
    batch_size: 160
    num_workers: 12
    pts_num: 4096
    load_from_preprocess: True

  OmniObject3d_val:
    root_dir: "/home/data/hofee/project/nbv_rec/data/sample_for_training_new"
    model_dir: "../data/scaled_object_meshes"
    source: nbv_reconstruction_dataset
    split_file: "/home/data/hofee/project/nbv_rec/data/sample.txt"
    type: test
    cache: True
    filter_degree: 75
    eval_list:
      - pose_diff
    ratio: 0.005
    batch_size: 160
    num_workers: 12
    pts_num: 4096
    load_from_preprocess: True


pipeline:
  nbv_reconstruction_local_pts_pipeline:
    modules:
      pts_encoder: pointnet_encoder
      seq_encoder: transformer_seq_encoder
      pose_encoder: pose_encoder
      view_finder: gf_view_finder
    eps: 1e-5
    global_scanned_feat: True

  nbv_reconstruction_global_pts_pipeline:
    modules:
      pts_encoder: pointnet_encoder
      pose_seq_encoder: transformer_seq_encoder
      pose_encoder: pose_encoder
      view_finder: gf_view_finder
    eps: 1e-5
    global_scanned_feat: True

  nbv_reconstruction_global_pts_n_num_pipeline:
    modules:
      pts_encoder: pointnet_encoder
      transformer_seq_encoder: transformer_seq_encoder
      pose_encoder: pose_encoder
      view_finder: gf_view_finder
      pts_num_encoder: pts_num_encoder
    eps: 1e-5
    global_scanned_feat: True

    
module:

  pointnet_encoder:
    in_dim: 3
    out_dim: 1024
    global_feat: True
    feature_transform: False

  transformer_seq_encoder:
    embed_dim: 384
    num_heads: 4
    ffn_dim: 256
    num_layers: 3
    output_dim: 2048

  gf_view_finder:
    t_feat_dim: 128
    pose_feat_dim: 256
    main_feat_dim: 3072
    regression_head: Rx_Ry_and_T
    pose_mode: rot_matrix
    per_point_feature: False
    sample_mode: ode
    sampling_steps: 500
    sde_mode: ve

  pose_encoder:
    pose_dim: 9
    out_dim: 256

  pts_num_encoder:
    out_dim: 64

loss_function:
  gf_loss:

evaluation_method:
  pose_diff:
  coverage_rate_increase: 
    renderer_path: "../blender/data_renderer.py"