config_train_occ.yaml

gpu: "0"
multi_gpu: "0,1" #null

model_name: model_difflow_occ
iters: 4

full: True

arch: PointConvLightRes
last_relu: False
allow_less_points: False

# general args
batch_size: 15
epochs: 560
learning_rate: 0.001

optimizer: AdamW
pretrain: ./pretrain_weights/model_difflow_occ_327_0.0428.pth
#pretrain: null
weight_decay: 0.0001
  
dim: 3
num_points: 8192

DEVICE: cuda

evaluate: False

dataset: FlyingThings3DSubset_Occlusion 
val_dataset: FlyingThings3DSubset_Occlusion # KITTI_Occlusion 
data_root: /dataset
val_path: /dataset

data_process:
  DEPTH_THRESHOLD: 35.
  NO_CORR: True

# data augmentation
aug_together:
  degree_range: 0.1745329252
  shift_range: 1.
  scale_low: 0.95
  scale_high: 1.05
  jitter_sigma: 0.01
  jitter_clip: 0.00

aug_pc2:
  degree_range: 0.
  shift_range: 0.3
  jitter_sigma: 0.01
  jitter_clip: 0.00

# miscalleneous
workers: 16

# uncertainty
uncertainty: 0.02