update codes

This commit is contained in:
D-X-Y
2019-02-01 17:17:02 +11:00
parent ea4d245a0e
commit 47327dc5a2
4 changed files with 12 additions and 10 deletions

View File

@@ -219,8 +219,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, args.dataset, args.data_path, args, basemodel.genotype(), 36, 20, log)
#torch.cuda.empty_cache()
#main_procedure(config, args.dataset, args.data_path, args, basemodel.genotype(), 36, 20, log)
log.close()