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

Import get_parameters from tinygrad.nn (#559)

Browse Source 
* get_parameter is in optim

* Update all imports for get_parameters

* Clean up

* use optim.get_paramters
This commit is contained in:
Jacky Lee
2023-02-17 15:22:26 -08:00
committed by GitHub
parent fae7654924
commit 9fd41632c6
13 changed files with 32 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_optim.py
View File

@@ -2,8 +2,7 @@ import numpy as np
import torch
import unittest
from tinygrad.tensor import Tensor
from tinygrad.nn.optim import Adam, SGD, RMSprop
from extra.utils import get_parameters
from tinygrad.nn.optim import Adam, SGD, RMSprop, get_parameters
x_init = np.random.randn(1,3).astype(np.float32)
W_init = np.random.randn(3,3).astype(np.float32)