def cool(self, epoch, logs): K.set_value( self.tau, np.max([self.min, self.max * np.exp(- self.anneal_rate * epoch)]))