Module: tfp.optimizer.convergence_criteria

View source on GitHub

TensorFlow Probability convergence criteria for optimizations.

Classes

class ConvergenceCriterion: Base class for stopping rules.

class LossNotDecreasing: Simple convergence criterion based on lack of decrease in loss values.

class SuccessiveGradientsAreUncorrelated: Convergence criterion based on inner products between successive gradients.