Skip to content

Commit d17257e

Browse files
committed
feat: change defaults for transform adapt
1 parent 3d845ac commit d17257e

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/sampler.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,8 +151,8 @@ impl Default for TransformedNutsSettings {
151151
Self {
152152
num_tune: 1500,
153153
num_draws: 1000,
154-
maxdepth: 8,
155-
max_energy_error: 1000f64,
154+
maxdepth: 10,
155+
max_energy_error: 20f64,
156156
store_gradient: false,
157157
store_unconstrained: false,
158158
store_divergences: false,

src/transform_adapt_strategy.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ pub struct TransformedSettings {
2323
impl Default for TransformedSettings {
2424
fn default() -> Self {
2525
Self {
26-
step_size_window: 0.1f64,
27-
transform_update_freq: 50,
26+
step_size_window: 0.07f64,
27+
transform_update_freq: 128,
2828
use_orbit_for_training: false,
29-
transform_train_max_energy_error: 50f64,
29+
transform_train_max_energy_error: 20f64,
3030
dual_average_options: Default::default(),
3131
}
3232
}

0 commit comments

Comments
 (0)