Update VIS-CODES and SCRIPTS

This commit is contained in:
D-X-Y 2020-07-23 08:01:21 +00:00
parent cb471a8966
commit b9fbe5577c

View File

@ -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)