melodyguan / enas

TensorFlow Code for paper "Efficient Neural Architecture Search via Parameter Sharing"
https://arxiv.org/abs/1802.03268
Apache License 2.0
1.58k stars 390 forks source link

Micro child reduction layers #87

Open alexrakowski opened 5 years ago

alexrakowski commented 5 years ago

I am confused with how many reduction (pooling?) layers a micro model should have. Figure 4 in the paper shows 1 reduction layer per 1 convolutional block, whereas in the code:

MicroChild.__init__(): ... pool_distance = self.num_layers // 3 self.pool_layers = [pool_distance, 2 * pool_distance + 1]

which would indicate there are always only two such layers. Is that correct? @melodyguan @hyhieu