From e22a3745ff9a9e4bb7aefa3f58c3fe6c296a6340 Mon Sep 17 00:00:00 2001 From: LMZimmer <50441716+LMZimmer@users.noreply.github.com> Date: Mon, 30 Nov 2020 04:28:58 -0500 Subject: [PATCH] Allow conv_kernel_size 1 --- autoPyTorch/components/networks/image/convnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autoPyTorch/components/networks/image/convnet.py b/autoPyTorch/components/networks/image/convnet.py index 8b3344460..b87bc0fd0 100644 --- a/autoPyTorch/components/networks/image/convnet.py +++ b/autoPyTorch/components/networks/image/convnet.py @@ -79,7 +79,7 @@ def get_config_space(user_updates=None): num_layers = CSH.UniformIntegerHyperparameter('num_layers', lower=2, upper=5) cs.add_hyperparameter(num_layers) cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_init_filters', lower=16, upper=64)) - cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_size', lower=2, upper=5)) + cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_size', lower=1, upper=5)) cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_stride', lower=1, upper=3)) cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_padding', lower=2, upper=3)) cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('pool_size', lower=2, upper=3))