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

added RMSprop optim

Browse Source
This commit is contained in:
f0ti
2020-10-23 02:50:02 +02:00
parent 65f3a9d499
commit 7e1eddb0c5
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_mnist.py
View File

@@ -44,7 +44,8 @@ class TestMNIST(unittest.TestCase):
steps = 400
else:
model = TinyBobNet()
optim = tinygrad_optim.SGD([model.l1, model.l2], lr=0.001)
# optim = tinygrad_optim.SGD([model.l1, model.l2], lr=0.001)
optim = tinygrad_optim.RMSprop([model.l1, model.l2], lr=0.001)
steps = 1000
BS = 128