nbv_grasping/configs/server_gsnet_preprocess_config.yaml
2024-10-09 16:13:22 +00:00

28 lines
565 B
YAML
Executable File

# Preprocess config file
settings:
general:
seed: 0
cuda_visible_devices: 0,1,2,3,4,5,6,7
device: cuda
test_dir: ""
experiment:
name: "new_gsnet_full_preprocess_test"
root_dir: "experiments"
keep_exists: False
preprocess:
dataset_list:
- source: "nbv1"
data_type: "test"
data_dir: "../data"
source: nbv1
batch_size: 1
scene_pts_num: 15000
voxel_size: 0.005
model:
general:
seed_feat_dim: 512
checkpoint_path: "../weights/epoch10.tar"