From 8af8a5f6ede44600156398546f58097b10b8bdd5 Mon Sep 17 00:00:00 2001 From: Kyurae Kim Date: Mon, 23 Oct 2023 23:48:20 -0400 Subject: [PATCH] fix documentation for `optimize` --- src/optimize.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/optimize.jl b/src/optimize.jl index 208ffabe..1a39cb41 100644 --- a/src/optimize.jl +++ b/src/optimize.jl @@ -35,7 +35,7 @@ The variational approximation can be constructed by passing the variational para - `optimizer::Optimisers.AbstractRule`: Optimizer used for inference. (Default: `Adam`.) - `rng::AbstractRNG`: Random number generator. (Default: `Random.default_rng()`.) - `show_progress::Bool`: Whether to show the progress bar. (Default: `true`.) -- `callback!`: Callback function called after every iteration. See further information below. (Default: `nothing`.) +- `callback`: Callback function called after every iteration. See further information below. (Default: `nothing`.) - `prog`: Progress bar configuration. (Default: `ProgressMeter.Progress(n_max_iter; desc="Optimizing", barlen=31, showspeed=true, enabled=prog)`.) - `state::NamedTuple`: Initial value for the internal state of optimization. Used to warm-start from the state of a previous run. (See the returned values below.)