从保存状态继续在pyro中进行MCMC(螺母)取样

我用他们的坚果采样器在pyro ppl中运行MCMC。我不能把所有的样品都存起来。现在我想在内存中作为缓冲区进行采样,将采样写入磁盘,然后从最后一个状态继续采样。

这可能吗?

我的内存中采样片段:

nuts_kernel = NUTS(pp_model, max_tree_depth=4,)
posterior = MCMC(nuts_kernel,
                num_samples=2000,
                warmup_steps=500).run(x_train, y_train)