Add preprocessing batch size for COVID19 dataset

• When loading the COVID19 datasets, we can now specify the batch size to use when we preprocess it. This is to avoid the memory exhaustion due to huge tensors.