a bug related to CNN search?

Hi, thanks for sharing the code. I am implementing the CNN part. I think the` block_idx` in `forward` function should be moded by 2 when CNN case as you used only two softmax. Could you check it? Thanks.

https://github.com/carpedm20/ENAS-pytorch/blob/25c4a89e17851d72e85566e16c81f0fd3749d58f/models/controller.py#L173-L177

Start with models/controller.py, especially the forward function at lines 173-177 linked in the issue, and trace the CNN case around block_idx and the two softmaxes. Confirm whether the index needs to be reduced modulo 2; done means CNN search uses the correct softmax index without changing other cases.

Written by the indexing model from the issue text.