From 9bf67ee812f752e282fc81ad85c6b44c8a5db287 Mon Sep 17 00:00:00 2001 From: Gregor von Laszewski Date: Sun, 15 Oct 2023 10:46:30 -0400 Subject: [PATCH] update config access --- benchmarks/cloudmask/target/summit-no-sciml/slstr_cloud.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/benchmarks/cloudmask/target/summit-no-sciml/slstr_cloud.py b/benchmarks/cloudmask/target/summit-no-sciml/slstr_cloud.py index d11fbcfa..01c3b8a4 100644 --- a/benchmarks/cloudmask/target/summit-no-sciml/slstr_cloud.py +++ b/benchmarks/cloudmask/target/summit-no-sciml/slstr_cloud.py @@ -193,6 +193,8 @@ def main(): with open(configFile, 'r') as stream: args = yaml.safe_load(stream) log_file = os.path.expanduser(args['log_file']) + + print(args) # Training start = time.time() @@ -221,7 +223,7 @@ def main(): StopWatch.event(f"number of inferences {number_inferences}") StopWatch.event(f"time per inference {time_per_inference}") StopWatch.event(f"nodes {args['nodes']}") - StopWatch.event(f"gpus {args['gpus']}") + StopWatch.event(f"gpus {args['gpu']}") print("number_inferences: ", number_inferences) with open(log_file, "a") as logfile: logfile.write(f"CloudMask inference, inferences={number_inferences}, bs={args['batch_size']}, nodes={args['nodes']}, gpus={args['gpu']}, time_per_inference={time_per_inference_str}\n")