Update VIS-CODES and SCRIPTS
This commit is contained in:
parent
cb471a8966
commit
b9fbe5577c
@ -459,7 +459,7 @@ def main(xargs):
|
|||||||
# the final post procedure : count the time
|
# the final post procedure : count the time
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
genotype, temp_accuracy = get_best_arch(valid_loader, network, xargs.eval_candidate_num, xargs.algo)
|
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)
|
network.set_cal_mode('dynamic', genotype)
|
||||||
elif xargs.algo == 'gdas':
|
elif xargs.algo == 'gdas':
|
||||||
network.set_cal_mode('gdas', None)
|
network.set_cal_mode('gdas', None)
|
||||||
|
Loading…
Reference in New Issue
Block a user