From f35553a0605198a251100fbf27952d01dad05e7a Mon Sep 17 00:00:00 2001 From: Peter Steinbach Date: Wed, 4 Mar 2020 14:48:23 +0100 Subject: [PATCH] top layer is ML lingo that requires an explanation for newcomers --- tutorials/01-basics/pytorch_basics/main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py index 744400c2..03e83c5f 100644 --- a/tutorials/01-basics/pytorch_basics/main.py +++ b/tutorials/01-basics/pytorch_basics/main.py @@ -168,6 +168,8 @@ def __len__(self): param.requires_grad = False # Replace the top layer for finetuning. +# top layer (or fully connected layer) refers to +# the last layer before the output of the network resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is an example. # Forward pass.