File size: 348 Bytes
a58327a 91867af a58327a |
1 2 3 4 5 6 7 8 9 10 11 |
_target_: src.datamodules.focus_datamodule.FocusDataModule
data_dir: ${data_dir}/focus150 # data_dir is specified in config.yaml
csv_train_file: ${data_dir}/focus150/train_metadata.csv
csv_val_file: ${data_dir}/focus150/validation_metadata.csv
csv_test_file: ${data_dir}/focus150/test_metadata.csv
batch_size: 64
num_workers: 0
pin_memory: False
|