Batch Size Implications
Author:

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.
Tasks: Deep Learning Fundamentals
Task Categories: Deep Learning Fundamentals
Published: 10/11/23
Tags
batchsize
batch
pytorch
memory utilization
Loading...