iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update TEMPLATE of GITHUB

Browse Source
This commit is contained in:
D-X-Y
2019-12-23 17:06:14 +11:00
parent f7a38d9170
commit 34ff796e1a
86 changed files with 42 additions and 4672 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/models/cell_searchs/search_model_gdas.py
View File

@@ -11,7 +11,7 @@ from .genotypes import Structure
class TinyNetworkGDAS(nn.Module):
def __init__(self, C, N, max_nodes, num_classes, search_space):
def __init__(self, C, N, max_nodes, num_classes, search_space, affine=False, track_running_stats=True):
super(TinyNetworkGDAS, self).__init__()
self._C = C
self._layerN = N
@@ -29,7 +29,7 @@ class TinyNetworkGDAS(nn.Module):
if reduction:
cell = ResNetBasicblock(C_prev, C_curr, 2)
else:
cell = SearchCell(C_prev, C_curr, 1, max_nodes, search_space)
cell = SearchCell(C_prev, C_curr, 1, max_nodes, search_space, affine, track_running_stats)
if num_edge is None: num_edge, edge2index = cell.num_edges, cell.edge2index
else: assert num_edge == cell.num_edges and edge2index == cell.edge2index, 'invalid {:} vs. {:}.'.format(num_edge, cell.num_edges)
self.cells.append( cell )