khanrc / pt.darts

PyTorch Implementation of DARTS: Differentiable Architecture Search
MIT License
439 stars 108 forks source link

featuremap size reduce in op.py FactorizedReduce function #18

Open yjump opened 5 years ago

yjump commented 5 years ago

Hi all, Thank you very much for this repo. Here I have a question. In op.py line 178 FactorizedReduce function, why conv2 layer need to reduce size on dim=2 and 3? In my case this cause dimension mismatch. out = torch.cat([self.conv1(x), self.conv2(x[:, :, 1:, 1:])], dim=1)