iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update affines for NAS

Browse Source
This commit is contained in:
D-X-Y
2019-12-02 18:03:40 +11:00
parent 487fec21bf
commit d175a361bd
9 changed files with 78 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/models/cell_infers/cells.py
View File

@@ -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 )

2
lib/models/cell_infers/tiny_network.py
View File

@@ -22,7 +22,7 @@ class TinyNetwork(nn.Module):
self.cells = nn.ModuleList()
for index, (C_curr, reduction) in enumerate(zip(layer_channels, layer_reductions)):
if reduction:
cell = ResNetBasicblock(C_prev, C_curr, 2)
cell = ResNetBasicblock(C_prev, C_curr, 2, True)
else:
cell = InferCell(genotype, C_prev, C_curr, 1)
self.cells.append( cell )