From 33533918911d46c9bcc2d98c68e5fd4b5098ad97 Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Fri, 1 Feb 2019 04:05:09 +1100 Subject: [PATCH] update codes --- exps-cnn/DARTS-Search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/exps-cnn/DARTS-Search.py b/exps-cnn/DARTS-Search.py index 9f2a88a..c6d6d65 100644 --- a/exps-cnn/DARTS-Search.py +++ b/exps-cnn/DARTS-Search.py @@ -222,8 +222,8 @@ def main(): print_log('Finish with training time = {:}'.format( convert_secs2time(total_train_time, True) ), log) # clear GPU cache - torch.cuda.empty_cache() - main_procedure(config, 'cifar10', os.environ['TORCH_HOME'] + '/cifar.python', args, basemodel.genotype(), 36, 20, log) + #torch.cuda.empty_cache() + #main_procedure(config, 'cifar10', os.environ['TORCH_HOME'] + '/cifar.python', args, basemodel.genotype(), 36, 20, log) log.close()