metric collector revert gpu arg name

nod-ai · Nov 26, 2024 · aa96f2f · aa96f2f
1 parent 0a4d628
commit aa96f2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ts/metrics/metric_collector.py b/ts/metrics/metric_collector.py
@@ -15,7 +15,7 @@
     parser = argparse.ArgumentParser()
 
     parser.add_argument(
-        "--gpus",
+        "--gpu",
         action="store",
         help="number of GPUs",
         type=int
@@ -24,6 +24,6 @@
 
     logging.basicConfig(stream=sys.stdout, format="%(message)s", level=logging.INFO)
 
-    system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpus)
+    system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpu)
 
     check_process_mem_usage(sys.stdin)