configs/dataset/osv5m.yaml

defaults:
  - train_transform: fast_clip
  - test_transform: fast_clip
  - _self_

name: osv5m
global_batch_size: 256

train_dataset:
  _partial_: true
  _target_: data.data.osv5m
  path: ${data_dir}/osv5m/
  split: train
  class_name: ${class_name}
  transforms: ${dataset.train_transform}
  aux_data: ${aux_data}
  is_baseline: ${is_baseline}
  areas: ${areas}
  streetclip: ${streetclip}
  blur: ${blur}

val_dataset:
  _partial_: true
  _target_: data.data.osv5m
  path: ${data_dir}/osv5m/
  split: val
  class_name: ${class_name}
  transforms: ${dataset.test_transform}
  aux_data: ${aux_data}
  is_baseline: ${is_baseline}
  areas: ${areas}
  streetclip: ${streetclip}
  blur: ${blur}

test_dataset:
  _partial_: true
  _target_: data.data.osv5m
  path: ${data_dir}/osv5m/
  split: test
  class_name: ${class_name}
  transforms: ${dataset.test_transform}
  aux_data: ${aux_data}
  is_baseline: ${is_baseline}
  areas: ${areas}
  streetclip: ${streetclip}
  blur: ${blur}