update affines for NAS

2019-12-02 18:03:40 +11:00
parent 487fec21bf
commit d175a361bd
9 changed files with 78 additions and 41 deletions
--- a/lib/models/cell_infers/cells.py
+++ b/lib/models/cell_infers/cells.py
@@ -19,9 +19,9 @@ class InferCell(nn.Module):
      cur_innod = []
      for (op_name, op_in) in node_info:
        if op_in == 0:
-          layer = OPS[op_name](C_in , C_out, stride)
+          layer = OPS[op_name](C_in , C_out, stride, True)
        else:
-          layer = OPS[op_name](C_out, C_out,      1)
+          layer = OPS[op_name](C_out, C_out,      1, True)
        cur_index.append( len(self.layers) )
        cur_innod.append( op_in )
        self.layers.append( layer )