search

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

2 bugs for macro_search #52

Open YinYangOfDao opened 6 years ago

YinYangOfDao commented 6 years ago
  1. in cifar10_macro_search.sh, the "child_num_cell_layers" should be "child_num_cells"
  2. in general_child.py/_model function, start_idx = self.num_branches is not correct. When searching partial channels, start_idx should also be 0 since in the controller there is no seqs inserted into the arc_seq/self.sample_arc