Продолжить выборку MCMC (NUTS) в пиро из сохраненного состояния

Question

Я использую MCMC в pyro-ppl с их сэмплером NUTS. Я не могу сохранить все образцы в памяти. Теперь я хочу сэмплировать в памяти как буфер, записать сэмплы на диск и продолжить сэмплирование с последнего состояния.

Это возможно?

Мой фрагмент памяти в памяти:

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