NAO_pytorch
NAO_pytorch copied to clipboard
Missing ReLU before Classifier
I could be wrong, but shouldn't there be a ReLU before the global pooling in NASNetworkCIFAR and NASNetworkImageNet? The tensorflow version of this code seems to have one:
https://github.com/renqianluo/NAO/blob/master/NAO/cnn/model.py#L682