Another fix

d2l-ai · Dec 8, 2022 · c44ccae · c44ccae
1 parent 6f65513
commit c44ccae
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/chapter_hyperparameter_optimization/sh-intro.md b/chapter_hyperparameter_optimization/sh-intro.md
@@ -130,7 +130,7 @@ def suggest(self):
         # Number of configurations for the first rung:
         n0 = int(self.prefact * self.eta ** self.K)
         for _ in range(n0):
-            config = searcher.sample_configuration()
+            config = self.searcher.sample_configuration()
             config["max_epochs"] = self.r_min  # Set r = r_min
             self.queue.append(config)
     # Return an element from the queue

diff --git a/d2l/torch.py b/d2l/torch.py
@@ -2735,7 +2735,7 @@ def suggest(self):
             # Number of configurations for the first rung:
             n0 = int(self.prefact * self.eta ** self.K)
             for _ in range(n0):
-                config = searcher.sample_configuration()
+                config = self.searcher.sample_configuration()
                 config["max_epochs"] = self.r_min  # Set r = r_min
                 self.queue.append(config)
         # Return an element from the queue
@@ -2784,7 +2784,7 @@ def __init__(self, searcher, eta, r_min, r_max):
         self.save_hyperparameters()
         self.s_max = int(np.ceil((np.log(r_max) - np.log(r_min)) / np.log(eta)))
         self.s = self.s_max
-        self.successive_halving = SuccessiveHalvingScheduler(
+        self.successive_halving = d2l.SuccessiveHalvingScheduler(
             searcher=self.searcher,
             eta=self.eta,
             r_min=self.r_min,
@@ -2806,7 +2806,7 @@ def update(self, config: dict, error: float, info=None):
             self.s -= 1
             if self.s < 0:
                 self.s = self.s_max
-            self.successive_halving = SuccessiveHalvingScheduler(
+            self.successive_halving = d2l.SuccessiveHalvingScheduler(
                 searcher=self.searcher,
                 eta=self.eta,
                 r_min=int(self.r_max * self.eta ** (-self.s)),