In this simple pytorch network with dataset and data loader our goal is to emphasize the significance of batch size and how it can improve performance like accuracy and also impact memory consumption or GPU utilization. We will use few different batch sizes in the training loop and plot the metrics for each batch size as a way to compare.