chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-04 18:00:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge de5bf73aa4 into e8baa81d89

Browse Source
This commit is contained in:
Mehnaaz 2024-10-14 20:00:33 +02:00 committed by GitHub
commit fbab48042d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
04_mnist_basics.ipynb
View File

@ -4825,10 +4825,10 @@
"class BasicOptim:\n",
" def __init__(self,params,lr): self.params,self.lr = list(params),lr\n",
"\n",
" def step(self, *args, **kwargs):\n",
" def step(self):\n",
" for p in self.params: p.data -= p.grad.data * self.lr\n",
"\n",
" def zero_grad(self, *args, **kwargs):\n",
" def zero_grad(self):\n",
" for p in self.params: p.grad = None"
]
},