update codes

2020-01-12 01:42:17 +11:00
parent 654015bf9d
commit 33384a78af
15 changed files with 288 additions and 21 deletions
--- a/lib/models/cell_searchs/search_model_enas.py
+++ b/lib/models/cell_searchs/search_model_enas.py
@@ -7,7 +7,7 @@ import torch
 import torch.nn as nn
 from copy import deepcopy
 from ..cell_operations import ResNetBasicblock
-from .search_cells     import SearchCell
+from .search_cells     import NAS102SearchCell as SearchCell
 from .genotypes        import Structure
 from .search_model_enas_utils import Controller