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

update GDAS

Browse Source
This commit is contained in:
D-X-Y
2020-01-18 21:54:17 +11:00
parent fcb6007975
commit 28d354880c
4 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/tf_models/cell_searchs/search_cells.py
View File

@@ -7,10 +7,10 @@ from copy import deepcopy
from ..cell_operations import OPS
class SearchCell(tf.keras.layers.Layer):
class NAS201SearchCell(tf.keras.layers.Layer):
def __init__(self, C_in, C_out, stride, max_nodes, op_names, affine=False):
super(SearchCell, self).__init__()
super(NAS201SearchCell, self).__init__()
self.op_names = deepcopy(op_names)
self.max_nodes = max_nodes

2
lib/tf_models/cell_searchs/search_model_gdas.py
View File

@@ -5,7 +5,7 @@ import tensorflow as tf
import numpy as np
from copy import deepcopy
from ..cell_operations import ResNetBasicblock
from .search_cells import SearchCell
from .search_cells import NAS201SearchCell as SearchCell
def sample_gumbel(shape, eps=1e-20):