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

ConvNeXt + extras (#657)

Browse Source 
* simple convnext implementation

* shorter function names

* need to realize the random functions now

* creating an optimizer realizes all params

* assign contiguous

* fix lazy lazy

* why was i doing that...add convnext to tests

* LazyNumpyArray

* enable assert + comment

* no two tiny
This commit is contained in:
George Hotz
2023-03-06 22:10:56 -08:00
committed by GitHub
parent d8dda2af3a
commit b14d31d6db
9 changed files with 111 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_tensor.py
View File

@@ -149,9 +149,9 @@ class TestTinygrad(unittest.TestCase):
for random_fn in [Tensor.randn, Tensor.uniform, Tensor.scaled_uniform, Tensor.glorot_uniform]:
with self.subTest(msg=f"Tensor.{random_fn.__name__}"):
Tensor.manual_seed(1337)
a = random_fn(10,10)
a = random_fn(10,10).realize()
Tensor.manual_seed(1337)
b = random_fn(10,10)
b = random_fn(10,10).realize()
np.testing.assert_allclose(a.numpy(), b.numpy())
if __name__ == '__main__':