Open neouyghur opened 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.
block_idx
forward
https://github.com/carpedm20/ENAS-pytorch/blob/25c4a89e17851d72e85566e16c81f0fd3749d58f/models/controller.py#L173-L177
Hi, thanks for sharing the code. I am implementing the CNN part. I think the
block_idx
inforward
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