From b9fbe5577ca2ce136921f82279b6d51cec961ca1 Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Thu, 23 Jul 2020 08:01:21 +0000
Subject: [PATCH] Update VIS-CODES and SCRIPTS

---
 exps/algos-v2/search-cell.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exps/algos-v2/search-cell.py b/exps/algos-v2/search-cell.py
index 7804cfa..976bc0a 100644
--- a/exps/algos-v2/search-cell.py
+++ b/exps/algos-v2/search-cell.py
@@ -459,7 +459,7 @@ def main(xargs):
   # the final post procedure : count the time
   start_time = time.time()
   genotype, temp_accuracy = get_best_arch(valid_loader, network, xargs.eval_candidate_num, xargs.algo)
-  if xargs.algo == 'setn':
+  if xargs.algo == 'setn' or xargs.algo == 'enas':
     network.set_cal_mode('dynamic', genotype)
   elif xargs.algo == 'gdas':
     network.set_cal_mode('gdas', None)