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

26 lines
492 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: "rgb_feat_preprocessor_test"
root_dir: "experiments"
keep_exists: True
preprocess:
dataset_list:
- source: "nbv1"
data_type: "sample"
data_dir: "../data"
source: nbv1
batch_size: 128
image_size: 480
model:
general:
model_name: "dinov2_vits14"