diff --git a/docs/src/callbacks.md b/docs/src/callbacks.md index bbf7783e..0156f238 100644 --- a/docs/src/callbacks.md +++ b/docs/src/callbacks.md @@ -122,7 +122,7 @@ cb = mcmc_callback( name_filter=( include=["mu", "sigma"], # Only log these parameters exclude=["_internal"], # Exclude matching names - stats=true, # Include step-level statistics + stats=false, # Include step-level statistics extras=true, # Include extra diagnostics ), ) @@ -143,8 +143,8 @@ cb = mcmc_callback( stats_options=(skip=50, thin=2), name_filter=( exclude=["_internal"], + stats=false, extras=true, - hyperparams=true, ), ) diff --git a/src/callbacks.jl b/src/callbacks.jl index 393f54a6..9c43b17a 100644 --- a/src/callbacks.jl +++ b/src/callbacks.jl @@ -278,7 +278,7 @@ Create a TensorBoard logging callback. **Requires TensorBoardLogger.jl to be loa - `true` or `:default`: Use default statistics (Mean, Variance, KHist) - requires OnlineStats - An OnlineStat or tuple of OnlineStats - requires OnlineStats - `stats_options`: NamedTuple with `thin`, `skip`, `window` -- `name_filter`: NamedTuple with `include`, `exclude`, `stats`, `hyperparams` +- `name_filter`: NamedTuple with `include`, `exclude`, `stats`, `extras` # Examples ```julia