data:
  batch_size: 4 
  shuffle: True
  num_workers: 4
  pin_memory: True