# Train config file

settings:
  general:
    seed: 0
    device: cuda
    cuda_visible_devices: "0,1,2,3,4,5,6,7"
    parallel: True
    
  experiment:
    name: train_experiment
    root_dir: "experiments"
    use_checkpoint: True
    epoch: -1 # -1 stands for last epoch
    max_epochs: 5000
    save_checkpoint_interval: 1
    test_first: True       
      
  train:
    optimizer:
      type: adam
      lr: 0.0001
    losses: # loss type : weight
      gf_loss: 1.0
    dataset:
      name: synthetic_train_train_dataset
      source: nbv1
      data_type: train
      ratio: 0.1
      batch_size: 128
      num_workers: 96

  test:
    batch_size: 16
    frequency: 3
    dataset_list:
      - name: synthetic_test_train_dataset
        source: nbv1
        data_type: train
        eval_list:
        ratio: 0.00001
        batch_size: 16
        num_workers: 16

  pipeline:
    pts_encoder: pointnet
    view_finder: gradient_field

datasets:
  general:
    data_dir: "../data"

modules:
  general:
    pts_channels: 3
    feature_dim: 1024
    per_point_feature: False
  pts_encoder:
    pointnet:
    pointnet++:
      params_name: light
    pointnet++rgb:
      params_name: light
      target_layer: 3
      rgb_feat_dim: 384
  view_finder:
    gradient_field:
      pose_mode: rot_matrix
      regression_head: Rx_Ry
      sample_mode: ode
      sample_repeat: 50
      sampling_steps: 500
      sde_mode: ve