diff --git a/TrainerClass.py b/TrainerClass.py index 2d0e8e7..b778d51 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -241,7 +241,7 @@ class eNoseTrainer: } total_space = (3*3*3*2*3*3*3*3) - self.logger.debug(f"total_space: {total_space}, num_samples: {int(nsamples*total_space)}") + self.logger.info(f"total_space: {total_space}, num_samples: {int(nsamples*total_space)}") scheduler = ASHAScheduler(metric='mse', mode='min', max_t=epochs, grace_period=5, reduction_factor=2) # analysis = tune.run(train_model, config=config_space, num_samples=num_samples, scheduler=scheduler)