daviddoji/Fastbook_Julia
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

store_attr

Browse Source
This commit is contained in:
Jeremy Howard 2020-09-13 09:55:40 -07:00
parent c6322c68a5
commit 45d5cd1c47
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
19_learner.ipynb
View File

@ -1326,7 +1326,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"class SGD:\n", "class SGD:\n",
" def __init__(self, params, lr, wd=0.): store_attr(self, 'params,lr,wd')\n", " def __init__(self, params, lr, wd=0.): store_attr()\n",
" def step(self):\n", " def step(self):\n",
" for p in self.params:\n", " for p in self.params:\n",
" p.data -= (p.grad.data + p.data*self.wd) * self.lr\n", " p.data -= (p.grad.data + p.data*self.wd) * self.lr\n",
@ -1367,7 +1367,7 @@
"source": [ "source": [
"class Learner:\n", "class Learner:\n",
" def __init__(self, model, dls, loss_func, lr, cbs, opt_func=SGD):\n", " def __init__(self, model, dls, loss_func, lr, cbs, opt_func=SGD):\n",
" store_attr(self, 'model,dls,loss_func,lr,cbs,opt_func')\n", " store_attr()\n",
" for cb in cbs: cb.learner = self\n", " for cb in cbs: cb.learner = self\n",
"\n", "\n",
" def one_batch(self):\n", " def one_batch(self):\n",

4
clean/19_learner.ipynb
View File

@ -606,7 +606,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"class SGD:\n", "class SGD:\n",
" def __init__(self, params, lr, wd=0.): store_attr(self, 'params,lr,wd')\n", " def __init__(self, params, lr, wd=0.): store_attr()\n",
" def step(self):\n", " def step(self):\n",
" for p in self.params:\n", " for p in self.params:\n",
" p.data -= (p.grad.data + p.data*self.wd) * self.lr\n", " p.data -= (p.grad.data + p.data*self.wd) * self.lr\n",
@ -633,7 +633,7 @@
"source": [ "source": [
"class Learner:\n", "class Learner:\n",
" def __init__(self, model, dls, loss_func, lr, cbs, opt_func=SGD):\n", " def __init__(self, model, dls, loss_func, lr, cbs, opt_func=SGD):\n",
" store_attr(self, 'model,dls,loss_func,lr,cbs,opt_func')\n", " store_attr()\n",
" for cb in cbs: cb.learner = self\n", " for cb in cbs: cb.learner = self\n",
"\n", "\n",
" def one_batch(self):\n", " def one_batch(self):\n",