Yes. I have two losses. Each loss take 16 images as a batch. Then the total loss is sum of the two losses. The backward is called for the total loss. So, I am not sure the total loss is performing on batch size of 16 or 32 ( two batch from two losses). Are you clear my problem?