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

Fix create_head initialization in 15_arch_details.ipynb (#383)

Browse Source 
Fixed as per the solution in https://forums.fast.ai/t/chapter-15-fastai-book-issue-siamese-training-running-mean/84534
This commit is contained in:
Pedro Pereira 2021-01-17 18:07:21 +00:00 committed by GitHub
parent 3916b71bdf
commit 663431d1ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
15_arch_details.ipynb
View File

@ -350,7 +350,7 @@
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
"source": [ "source": [
"Then we can create our head. A look at the encoder tells us the last layer has 512 features, so this head will need to receive `512*4`. Why 4? First we have to multiply by 2 because we have two images. Then we need a second multiplication by 2 because of our concat-pool trick. So we create the head as follows:" "Then we can create our head. A look at the encoder tells us the last layer has 512 features, so this head will need to receive `512*2`. Why 2? We have to multiply by 2 because we have two images. So we create the head as follows:"
] ]
}, },
{ {
@ -359,7 +359,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"head = create_head(512*4, 2, ps=0.5)" "head = create_head(512*2, 2, ps=0.5)"
] ]
}, },
{ {