diff --git a/tutorials/03-bayesian-neural-network/index.qmd b/tutorials/03-bayesian-neural-network/index.qmd index fcfa7502f..87d556634 100755 --- a/tutorials/03-bayesian-neural-network/index.qmd +++ b/tutorials/03-bayesian-neural-network/index.qmd @@ -19,7 +19,7 @@ using Turing using FillArrays using Lux using Plots -using Tracker +import Mooncake using Functors using LinearAlgebra @@ -208,7 +208,7 @@ setprogress!(false) ```{julia} # Perform inference. N = 2_000 -ch = sample(bayes_nn(reduce(hcat, xs), ts), NUTS(; adtype=AutoTracker()), N); +ch = sample(bayes_nn(reduce(hcat, xs), ts), NUTS(; adtype=AutoMooncake(; config=nothing)), N); ``` Now we extract the parameter samples from the sampled chain as `θ` (this is of size `5000 x 20` where `5000` is the number of iterations and `20` is the number of parameters).