enas
enas copied to clipboard
Possible Bug in in micro_child.py
Hi,
It appears to me that while building the input to the next pooling layer, we are throwing away layers[0] element with this line. https://github.com/melodyguan/enas/blob/d1a90ac915301198f2a30ce136e9040e6c4235ff/src/cifar10/micro_child.py#L277
I think it should be instead layers = layers[layers[0], x]
Before the factorized_reduction
operation x would already contain the layers[-1]
element from previous loop and I think we should only replace the last element of layers
with the factorized_reduction(x)
.
Instead what is being done is: layers = [x, factorized_reduction(x)]
@melodyguan Hi, could you please have a look into this.
I am also curious about this problem. Hope to get response from author.