Я использую 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)