From 41825bf0ec2f8d4e916bea46cbdf6921686a88a3 Mon Sep 17 00:00:00 2001
From: Chao Peng <cpeng@anl.gov>
Date: Mon, 5 Dec 2022 14:42:58 -0600
Subject: [PATCH] add a run_type for run_benchmark.py
---
benchmarks/imaging_shower_ML/run_benchmark.py | 28 ++++++++++++-------
1 file changed, 18 insertions(+), 10 deletions(-)
diff --git a/benchmarks/imaging_shower_ML/run_benchmark.py b/benchmarks/imaging_shower_ML/run_benchmark.py
index bc91124a..a196fcbe 100755
--- a/benchmarks/imaging_shower_ML/run_benchmark.py
+++ b/benchmarks/imaging_shower_ML/run_benchmark.py
@@ -26,14 +26,14 @@ FILE_NAMES = dict(
prep_script = os.path.join(SDIR, 'scripts', 'ml_data_preprocess.py'),
ml_script = os.path.join(SDIR, 'scripts', 'ml_training.py'),
- sim_dir = os.path.join('{outdir}', 'sim_data'),
- epscan_dir = os.path.join('{outdir}', 'epcut'),
- ml_dir = os.path.join('{outdir}', 'ml_result'),
+ sim_dir = os.path.join('{outdir}', '{run_type}', 'sim_data'),
+ epscan_dir = os.path.join('{outdir}', '{run_type}', 'epcut'),
+ ml_dir = os.path.join('{outdir}', '{run_type}', 'ml_result'),
- gen_file = os.path.join('{outdir}', 'sim_data', '{ntag}_gen.hepmc'),
- sim_file = os.path.join('{outdir}', 'sim_data', '{ntag}_sim.edm4hep.root'),
- rec_file = os.path.join('{outdir}', 'sim_data', '{ntag}_rec.root'),
- ml_file = os.path.join('{outdir}', 'sim_data', '{ntag}_data.h5'),
+ gen_file = os.path.join('{outdir}', '{run_type}', 'sim_data', '{ntag}_gen.hepmc'),
+ sim_file = os.path.join('{outdir}', '{run_type}', 'sim_data', '{ntag}_sim.edm4hep.root'),
+ rec_file = os.path.join('{outdir}', '{run_type}', 'sim_data', '{ntag}_rec.root'),
+ ml_file = os.path.join('{outdir}', '{run_type}', 'sim_data', '{ntag}_data.h5'),
)
# default values for argument parser
DEFAULT_COMPACT = os.path.join(
@@ -124,11 +124,17 @@ if __name__ == '__main__':
default='sim_output',
help='output directory.'
)
+ parser.add_argument(
+ '-r', '--run-type', type=str,
+ dest='run_type',
+ default='epic_imcal',
+ help='a name specify the run type.'
+ )
parser.add_argument(
'-t', '--name-tag', type=str,
dest='ntag',
default='imcal_ml',
- help='name tag for output files.'
+ help='a name tag for output files.'
)
parser.add_argument(
'-c', '--compact', type=str,
@@ -271,11 +277,13 @@ if __name__ == '__main__':
# save run information
# NOTE: runs with the same outdir will be combined (overwrite if nametag is the same)
- run_data = {args.ntag: kwargs}
+ run_data = {args.run_type: {args.ntag: kwargs}}
try:
with open(os.path.join(args.outdir, 'result.json'), 'r') as f:
run_data = json.load(f)
- run_data.update({args.ntag: kwargs})
+ run_info = run_data.get(args.run_type, {})
+ run_info.update({args.ntag: kwargs})
+ run_data[args.run_type] = run_info
except (FileNotFoundError, json.decoder.JSONDecodeError):
pass
--
GitLab