chris/fastbook
1
0
Fork 0
mirror of https://github.com/fastai/fastbook.git synced 2025-04-07 19:30:43 +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", "class BasicOptim:\n",
" def __init__(self,params,lr): self.params,self.lr = list(params),lr\n", " def __init__(self,params,lr): self.params,self.lr = list(params),lr\n",
"\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", " for p in self.params: p.data -= p.grad.data * self.lr\n",
"\n", "\n",
" def zero_grad(self, *args, **kwargs):\n", " def zero_grad(self):\n",
" for p in self.params: p.grad = None" " for p in self.params: p.grad = None"
] ]
}, },