num_workers = 4
train_batch_size = 32
val_batch_size = 1