CUQI-DTU / CUQIpy

https://cuqi-dtu.github.io/CUQIpy/
Apache License 2.0
41 stars 7 forks source link

Allow choosing number of steps in warmup #460

Open nabriis opened 1 month ago

nabriis commented 1 month ago

Is your feature request related to a problem? Please describe.

One should be able to choose the number of steps for both warmup and sampling in HybridGibbsNew

import cuqi
num_steps_warmup = {
"x" : 100,
"s": 10
}
num_steps_sample = {
"x" : 10,
"s": 1
}

Describe the solution you'd like

See above.