Skip to content

Commit c81f9b7

Browse files
authored
Fix theta optim decision using q_points iter nb (#248)
1 parent aaf3c2a commit c81f9b7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

crates/ego/src/solver/solver_impl.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ where
126126
if make_clustering
127127
/* init || recluster */
128128
{
129-
info!("{} Clustering and training...", model_name);
129+
info!("{} clustering and training...", model_name);
130130
let model = builder
131131
.train(xt.view(), yt.view())
132132
.expect("GP training failure");
@@ -448,7 +448,7 @@ where
448448
};
449449
let make_clustering = (init && i == 0) || recluster;
450450
let optimize_theta =
451-
(iter as usize * self.config.q_points) % (self.config.n_optmod) == 0;
451+
(iter as usize * self.config.q_points + i) % (self.config.n_optmod) == 0;
452452
self.make_clustered_surrogate(
453453
&name,
454454
&xt,

0 commit comments

Comments
 (0)