mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix error message (#4556)

Browse source 
* fix error messgae

* typo

* add suggestion to fix error

---------

Co-authored-by: Thomas Ziereis <thomas.ziereis@web.de>
This commit is contained in:
ziereis 2024-05-12 21:35:51 +02:00 • committed by GitHub
commit bcee4743ce
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tinygrad/nn/optim.py
View file

@ -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