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

fix: 13_convolutions.ipynb - we should be normalizing activations (#366)

Browse Source
This commit is contained in:
Yurij Mikhalevich 2021-01-17 21:11:30 +03:00 committed by GitHub
parent 829b68ca05
commit b4eac5e921
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
13_convolutions.ipynb
View File

@ -3415,8 +3415,8 @@
"source": [ "source": [
"def conv(ni, nf, ks=3, act=True):\n", "def conv(ni, nf, ks=3, act=True):\n",
" layers = [nn.Conv2d(ni, nf, stride=2, kernel_size=ks, padding=ks//2)]\n", " layers = [nn.Conv2d(ni, nf, stride=2, kernel_size=ks, padding=ks//2)]\n",
" layers.append(nn.BatchNorm2d(nf))\n",
" if act: layers.append(nn.ReLU())\n", " if act: layers.append(nn.ReLU())\n",
" layers.append(nn.BatchNorm2d(nf))\n",
" return nn.Sequential(*layers)" " return nn.Sequential(*layers)"
] ]
}, },