batch_size=3 subsets_balanced Gpus=1 batch_size=