Closed rohandahale closed 1 year ago
Dev docs for example: https://ptiede.github.io/Comrade.jl/dev/examples/imaging_vis/
In chain, stats = sample(rng, post, AHMC(;metric, autodiff=AD.ZygoteBackend()), 400; nadapts=300, init_params=x0) , AHMC(;metric, autodiff=AD.ZygoteBackend()) does not work and should be changed to AHMC(;metric, autodiff=Val(:Zygote))
chain, stats = sample(rng, post, AHMC(;metric, autodiff=AD.ZygoteBackend()), 400; nadapts=300, init_params=x0)
AHMC(;metric, autodiff=AD.ZygoteBackend())
AHMC(;metric, autodiff=Val(:Zygote))
fixed
Dev docs for example: https://ptiede.github.io/Comrade.jl/dev/examples/imaging_vis/
In
chain, stats = sample(rng, post, AHMC(;metric, autodiff=AD.ZygoteBackend()), 400; nadapts=300, init_params=x0)
,AHMC(;metric, autodiff=AD.ZygoteBackend())
does not work and should be changed toAHMC(;metric, autodiff=Val(:Zygote))