more lint

This commit is contained in:
npapernot 2019-07-29 22:00:53 +00:00
parent ddf17c9091
commit 32c76e588a

View file

@ -40,12 +40,12 @@ class GammaBetaDecreasingStep(
def __call__(self, step): def __call__(self, step):
"""Computes and returns the learning rate. """Computes and returns the learning rate.
Args: Args:
step: the current iteration number step: the current iteration number
Returns: Returns:
decayed learning rate to minimum of 1/beta and 1/(gamma * step) as per decayed learning rate to minimum of 1/beta and 1/(gamma * step) as per
the Bolton privacy requirements. the Bolton privacy requirements.
""" """
if not self.is_init: if not self.is_init:
raise AttributeError('Please initialize the {0} Learning Rate Scheduler.' raise AttributeError('Please initialize the {0} Learning Rate Scheduler.'