output_node = input_node
kernel_size = utils.add_to_hp(self.kernel_size, hp)
num_blocks = self.num_blocks or hp.Choice("num_blocks", [1, 2, 3], default=2)
num_layers = self.num_layers or hp.Choice("num_layers", [1, 2], default=2)
separable = self.separable
if separable is None:
separable = hp.Boolean("separable", default=False)
After Change
else:
dropout = hp.Choice("dropout", [0.0, 0.25, 0.5], default=0)
for i in range(utils.add_to_hp(self.num_blocks, hp)):
for j in range(num_layers):
output_node = conv(
hp.Choice(
