add log message before training

lbl-camera · Aug 2, 2023 · 6ef81a6 · 6ef81a6
1 parent 8724b13
commit 6ef81a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tsuchinoko/adaptive/gpCAM_in_process.py b/tsuchinoko/adaptive/gpCAM_in_process.py
@@ -165,6 +165,7 @@ def train(self):
 
             for N in train_at:
                 if len(self.optimizer.y_data) > N and N not in self._completed_training[method]:
+                    logger.info('Training in progress. This make take a while...')
                     self.optimizer.train(np.asarray([[self.parameters[('hyperparameters', f'hyperparameter_{i}_{edge}')]
                                                       for edge in ['min', 'max']]
                                                      for i in range(self.num_hyperparameters)]),