github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-25 23:08:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: this makes beautiful mnist work again, not okay

Browse Source
This commit is contained in:
George Hotz
2024-03-18 18:22:44 -07:00
parent a6ed2ae3c6
commit e6d55932ca
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/nn/optim.py
View File

@@ -20,7 +20,10 @@ class Optimizer:
def realize(self, extra=None):
# NOTE: in extra is too late for most of the params due to issues with assign
Tensor.corealize(extra + self.params + self.buffers if extra is not None else self.params + self.buffers)
#Tensor.corealize(extra + self.params + self.buffers if extra is not None else self.params + self.buffers)
# TODO: SUPER BROKEN THAT THIS FIXES IT
if extra is not None: Tensor.corealize(extra)
Tensor.corealize(self.params + self.buffers)
def step(self) -> None: raise NotImplementedError