I am doing warmup in windows of certain intervals and I wish to obtain the time step size and the inverse mass matrix after these warmup windows. How can I do so in Numpyro?
there are several forum threads show you how, e.g. Get number of steps, sizes and acc. prob over all iterations
1 Like