cuqi.experimental.mcmc.UGLA.set_state

UGLA.set_state(state)

Set the state of the sampler.

The state is used when loading the sampler from a checkpoint.

The state of the sampler is a dictionary with keys ‘metadata’ and ‘state’.

For example, the state of a “MH” sampler could be:

state = {

‘metadata’: {

‘sampler_type’: ‘MH’

}, ‘state’: {

‘current_point’: np.array([…]), ‘current_target_logd’: -123.45, ‘scale’: 1.0, …

}

}