train: 
  dataset_dir: C:\Users\leob3\AGen\data\train
  batch_size: 1  
  shuffle: True   
  num_workers: 4  

valid: 
  dataset_dir: C:\Users\leob3\AGen\data\valid
  batch_size: 1  
  shuffle: True   
  num_workers: 4  

test: 
  dataset_dir: C:\Users\leob3\AGen\data\test
  batch_size: 1  
  shuffle: True   
  num_workers: 4