fix error message (#4556)

* fix error messgae * typo * add suggestion to fix error --------- Co-authored-by: Thomas Ziereis <[email protected]>
2024-05-12 12:35:51 -07:00
parent 01a0c1a948
commit bcee4743ce
1 changed files with 3 additions and 1 deletions
@@ -23,7 +23,9 @@ class Optimizer:

  def step(self): Tensor.realize(*self.schedule_step())
  def schedule_step(self) -> List[Tensor]:
-    assert Tensor.training, "Tensor.training should be set in the optimizer"
+    assert Tensor.training, (
+            f"""Tensor.training={Tensor.training}, Tensor.training must be enabled to use the optimizer.
+                - help: Consider setting Tensor.training=True before calling Optimizer.step().""")
    return self._step()+self.params+self.buffers
  def _step(self) -> List[Tensor]: raise NotImplementedError