update GDAS

2020-01-18 21:54:17 +11:00
parent fcb6007975
commit 28d354880c
4 changed files with 10 additions and 3 deletions
--- a/lib/tf_models/cell_searchs/search_model_gdas.py
+++ b/lib/tf_models/cell_searchs/search_model_gdas.py
@@ -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):