(self, lr)
Set the learning rate.
Set the learning rate.
Set the learning rate .
def set_lr(self, lr):
        """Set the learning rate."""
        for param_group in self.optimizer.param_groups:
            param_group['lr'] = lr
def set_lr ( self , lr ) : for param_group in self . optimizer . param_groups : param_group [ 'lr' ] = lr