ENAS-pytorch icon indicating copy to clipboard operation
ENAS-pytorch copied to clipboard

a bug related to CNN search?

Open neouyghur opened this issue 5 years ago • 0 comments

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

neouyghur avatar Dec 19 '19 05:12 neouyghur