def METHOD_NAME(self, epoch, ind_batch):
    """
    if hyper-parameters should be updated per batch, then step
    should be set to epoch*self.num_batches + ind_batch
    """
    if self.flag_update_hyper_per_batch:
        self.model.hyper_update(epoch*self.num_batches + ind_batch, self.hyper_scheduler)
    return super().METHOD_NAME(epoch, ind_batch)