Allow conv_kernel_size 1

franchuterivera · Nov 30, 2020 · e22a374 · e22a374
1 parent 2f7d829
commit e22a374
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))