sampling interval to config

undertherain · Sep 26, 2020 · 888cf12 · 888cf12
1 parent 0f6a017
commit 888cf12
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/benchmarker/modules/i_neural_net.py b/benchmarker/modules/i_neural_net.py
@@ -19,13 +19,16 @@ def __init__(self, params, extra_args=None):
         parser = argparse.ArgumentParser(description="Benchmark deep learning models")
         parser.add_argument("--mode", default="training")
         parser.add_argument("--nb_epoch", type=int, default=10)
+        parser.add_argument("--power_sampling_ms", type=int, default=100)
 
         #
         parser.add_argument("--random_seed", default=None)
 
         parsed_args, remaining_args = parser.parse_known_args(extra_args)
 
         params["mode"] = parsed_args.mode
+        params["power"] = {}
+        params["power"]["sampling_ms"] = parsed_args.power_sampling_ms
         params["nb_epoch"] = parsed_args.nb_epoch
         assert params["mode"] in ["training", "inference"]
         params["path_out"] = os.path.join(params["path_out"], params["mode"])
@@ -119,11 +122,8 @@ def monitor(self):
         handle = nvmlDeviceGetHandleByIndex(0)
 
         while self.keep_monitor:
-            # print("MONITOR")
             powDraw = (nvmlDeviceGetPowerUsage(handle) / 1000.0)
             power_gpu.append(powDraw)
-            # TODO: measurement interval to config
-            sleep(0.1)
+            sleep(self.params["power"]["sampling_ms"] / 1000.0)
         nvmlShutdown()
-        self.params["power"] = {}
         self.params["power"]["GPU"] = np.mean(power_gpu)