def createLearningRateScheduler(self, optimizer):
    """
    Creates the learning rate scheduler and attach the optimizer
    """
    return torch.optim.lr_scheduler.StepLR(optimizer,
                                           step_size=1,
                                           gamma=self.lr_scheduler_gamma)
Creates the learning rate scheduler and attach the optimizer