def _num_training_batches_reached(self, is_last_batch=False):
    return (((self.trainer.batch_idx + 1) == self.trainer.num_training_batches) or is_last_batch)
def _num_training_batches_reached(self, is_last_batch=False):
    return (((self.trainer.batch_idx + 1) == self.trainer.num_training_batches) or is_last_batch)