Skip to content

Commit e22a374

Browse files
authored
Allow conv_kernel_size 1
1 parent 2f7d829 commit e22a374

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autoPyTorch/components/networks/image/convnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def get_config_space(user_updates=None):
7979
num_layers = CSH.UniformIntegerHyperparameter('num_layers', lower=2, upper=5)
8080
cs.add_hyperparameter(num_layers)
8181
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_init_filters', lower=16, upper=64))
82-
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_size', lower=2, upper=5))
82+
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_size', lower=1, upper=5))
8383
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_stride', lower=1, upper=3))
8484
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('conv_kernel_padding', lower=2, upper=3))
8585
cs.add_hyperparameter(CSH.UniformIntegerHyperparameter('pool_size', lower=2, upper=3))

0 commit comments

Comments
 (0)