device: cuda
num_gpus: 1
num_workers: 8