iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update codes

Browse Source
This commit is contained in:
D-X-Y 2019-02-01 04:05:09 +11:00
parent 65d9c1c57f
commit 3353391891
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
exps-cnn/DARTS-Search.py
View File

@ -222,8 +222,8 @@ def main():
print_log('Finish with training time = {:}'.format( convert_secs2time(total_train_time, True) ), log) print_log('Finish with training time = {:}'.format( convert_secs2time(total_train_time, True) ), log)
# clear GPU cache # clear GPU cache
torch.cuda.empty_cache() #torch.cuda.empty_cache()
main_procedure(config, 'cifar10', os.environ['TORCH_HOME'] + '/cifar.python', args, basemodel.genotype(), 36, 20, log) #main_procedure(config, 'cifar10', os.environ['TORCH_HOME'] + '/cifar.python', args, basemodel.genotype(), 36, 20, log)
log.close() log.close()