我正在使用他们的NUTS采样器在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)