It seems like that the two losses computed for these two batches are equal and the second batch is much cheaper to process because it don’t need to compute the duplicate DOG samples. Am I right? So, how do you suggest to deal with this problem? Since it seems like that you don‘t’ like these two batches either.