def update_lr(self, names=None):
    names = self.get_model_names(names)
    for name in names:
        if (self._scheds[name] is not None):
            self._scheds[name].step()
def update_lr(self, names=None):
    names = self.get_model_names(names)
    for name in names:
        if (self._scheds[name] is not None):
            self._scheds[name].step()