ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.optimizer.convergence_criteria

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.