PyTorch modules do not depend on the batch size, so you could pick this value.
However, as is shown in a lot of posts in this forum, a too small batch size might yield skewed running stats and your model might perform poorly on the validation and test set. The GroupNorm paper explains this in more detail and proposes the nn.GroupNorm layer.