Skip to content

Commit bab0b62

Browse files
committed
feat: adapt some default parameters for transformed adaptation
1 parent c74a601 commit bab0b62

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sampler.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ impl Default for LowRankNutsSettings {
149149
impl Default for TransformedNutsSettings {
150150
fn default() -> Self {
151151
Self {
152-
num_tune: 4000,
152+
num_tune: 1500,
153153
num_draws: 1000,
154154
maxdepth: 8,
155155
max_energy_error: 1000f64,
@@ -159,7 +159,7 @@ impl Default for TransformedNutsSettings {
159159
adapt_options: Default::default(),
160160
check_turning: true,
161161
seed: 0,
162-
num_chains: 6,
162+
num_chains: 1,
163163
}
164164
}
165165
}

src/transform_adapt_strategy.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ impl Default for TransformedSettings {
2525
Self {
2626
step_size_window: 0.1f64,
2727
transform_update_freq: 50,
28-
use_orbit_for_training: true,
28+
use_orbit_for_training: false,
2929
transform_train_max_energy_error: 50f64,
3030
dual_average_options: Default::default(),
3131
}

0 commit comments

Comments
 (0)