I think the reduce param in the Loss functions might be helpful.

reduce (bool, optional) – By default, the losses are averaged or summed for each minibatch. When reduce is False, the loss function returns a loss per batch element instead and ignores size_average. Default: True

Using this, you could increase the batch size and get the losses for each batch.