73 lines
1.5 KiB
YAML
73 lines
1.5 KiB
YAML
# 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 |