Attend the Women in ML Symposium on December 7 Register now

Module: tfp.substrates.jax.optimizer.convergence_criteria

Stay organized with collections Save and categorize content based on your preferences.

TensorFlow Probability convergence criteria for optimizations.


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.