Skip to content

Commit bb1dec1

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

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sampler.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ impl Default for TransformedNutsSettings {
152152
num_tune: 1500,
153153
num_draws: 1000,
154154
maxdepth: 8,
155-
max_energy_error: 1000f64,
155+
max_energy_error: 50f64,
156156
store_gradient: false,
157157
store_unconstrained: false,
158158
store_divergences: false,

src/transform_adapt_strategy.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ impl Default for TransformedSettings {
2424
fn default() -> Self {
2525
Self {
2626
step_size_window: 0.1f64,
27-
transform_update_freq: 50,
27+
transform_update_freq: 128,
2828
use_orbit_for_training: false,
2929
transform_train_max_energy_error: 50f64,
3030
dual_average_options: Default::default(),

0 commit comments

Comments
 (0)