zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nits + rebase

Browse Source
This commit is contained in:
Awni Hannun 2024-01-01 21:28:58 -08:00
parent d2a826b3a4
commit bb41f14add
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/tests/test_nn.py
View File

@ -12,9 +12,6 @@ from mlx.utils import tree_flatten, tree_map, tree_unflatten
class TestBase(mlx_tests.MLXTestCase): class TestBase(mlx_tests.MLXTestCase):
def test_update(self):
pass
def test_module_utilities(self): def test_module_utilities(self):
m = nn.Sequential( m = nn.Sequential(
nn.Sequential(nn.Linear(2, 10), nn.relu), nn.Sequential(nn.Linear(2, 10), nn.relu),
@ -72,7 +69,7 @@ class TestBase(mlx_tests.MLXTestCase):
eq_tree = tree_map(mx.array_equal, m.parameters(), m_load.parameters()) eq_tree = tree_map(mx.array_equal, m.parameters(), m_load.parameters())
self.assertTrue(all(tree_flatten(eq_tree))) self.assertTrue(all(tree_flatten(eq_tree)))
def test_from_weights(self): def test_load_from_weights(self):
m = nn.Linear(2, 2) m = nn.Linear(2, 2)
# Too few weights # Too few weights