search

carpedm20 / ENAS-pytorch

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"
Apache License 2.0
2.69k stars 492 forks source link

a bug related to CNN search? #49

Open neouyghur opened 4 years ago

neouyghur commented 4 years ago

Hi, thanks for sharing the code. I am implementing the CNN part. I think theblock_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