Update zeus/optimizer/power_limit.py

Co-authored-by: Jae-Won Chung <jwnchung@umich.edu>
ml-energy · Feb 13, 2024 · 46f2a88 · 46f2a88
1 parent dc9c28f
commit 46f2a88
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zeus/optimizer/power_limit.py b/zeus/optimizer/power_limit.py
@@ -497,6 +497,8 @@ class HFGlobalPowerLimitOptimizer(TrainerCallback):
     """[Wrapped for Hugging Face Trainer Callback] Optimizer for the power limit knob.
 
     This optimizer uses the JIT profiling log to determine the optimal power limit.
+    See [`GlobalPowerLimitOptimizer`][zeus.optimizer.power_limit.GlobalPowerLimitOptimizer]
+    for the underlying optimizer implementation.
     """
 
     def __init__(